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