def define_D(opt): opt_net = opt['network_D'] which_model = opt_net['which_model_D'] if which_model == 'discriminator_vgg_128': netD = SRGAN_arch.Discriminator_VGG_128(in_nc=opt_net['in_nc'], nf=opt_net['nf']) else: raise NotImplementedError('Discriminator model [{:s}] not recognized'.format(which_model)) return netD
def define_D(opt): opt_net = opt['network_D'] which_model = opt_net['which_model_D'] if which_model == 'discriminator_vgg_128': netD = SRGAN_arch.Discriminator_VGG_128(in_nc=opt_net['in_nc'], nf=opt_net['nf']) elif which_model == 'discriminator_patch': norm_layer = get_norm_layer(norm_type=opt_net['Instance']) netD = SRGAN_patch_arch.NLayerDiscriminator(input_nc=opt_net['in_nc'], ndf=opt_net['nf'], n_layers=3, norm_layer=norm_layer) else: raise NotImplementedError( 'Discriminator model [{:s}] not recognized'.format(which_model)) return netD
def define_D(opt): opt_net = opt['network_D'] which_model = opt_net['which_model_D'] if which_model == 'discriminator_vgg_128': netD = SRGAN_arch.Discriminator_VGG_128(in_nc=opt_net['in_nc'], nf=opt_net['nf']) elif which_model == 'DCN3D': netD = SRGAN_arch.MultiscaleDiscriminator( input_nc=opt_net['input_nc'], ndf=opt_net['ndf'], n_layers=opt_net['n_layers'], num_D=opt_net['num_d'], use_sigmoid=opt_net['use_sigmoid'], get_inter_feat=opt_net['get_inter_feat'], has_bias=opt_net['has_bias'], has_sn=opt_net['has_sn'], max_ndf=opt_net['max_ndf'], conv_type=opt_net['conv_type']) else: raise NotImplementedError( 'Discriminator model [{:s}] not recognized'.format(which_model)) return netD