def find_network_using_name(target_network_name, filename): target_class_name = target_network_name + filename module_name = 'models.networks.' + filename network = util.find_class_in_module(target_class_name, module_name) assert issubclass(network, BaseNetwork), \ "Class %s should be a subclass of BaseNetwork" % network return network
def find_network_using_name(target_network_name, filename, extend=None): extend = filename if extend is None else extend target_class_name = target_network_name + extend module_name = 'model.networks.' + filename network = util.find_class_in_module(target_class_name, module_name) assert issubclass(network, BaseNetwork), \ "Class %s should be a subclass of BaseNetwork" % network return network
def modify_commandline_options(parser, is_train): parser.add_argument('--netD_subarch', type=str, default='n_layer', help='architecture of each discriminator') parser.add_argument('--num_D', type=int, default=2, help='number of discriminators to be used in multiscale') opt, _ = parser.parse_known_args() # define properties of each discriminator of the multiscale discriminator subnetD = util.find_class_in_module(opt.netD_subarch + 'discriminator', 'models.networks.discriminator') subnetD.modify_commandline_options(parser, is_train) return parser