def build_img_discriminator(args, vocab):
discriminator = None
d_kwargs = {}
d_weight = args.discriminator_loss_weight
d_img_weight = args.d_img_weight
if d_weight == 0 or d_img_weight == 0:
return discriminator, d_kwargs
d_kwargs = {
'arch': args.d_img_arch,
'normalization': args.d_normalization,
'activation': args.d_activation,
'padding': args.d_padding
}
if args.sg_context_dim_d > 0:
#discriminator has access to pooled GCNN features
layout_dim = 0
###########################
#discriminator has access to layout
if args.layout_for_discrim:
layout_dim = args.gconv_dim
if args.image_patch_discr:
discriminator = CondGANPatchDiscriminator(
**d_kwargs,
layout_dim=layout_dim,
sg_context_dim=args.sg_context_dim_d)
else:
discriminator = CondGANDiscriminator(
**d_kwargs,
layout_dim=layout_dim,
sg_context_dim=args.sg_context_dim_d)
else:
discriminator = PatchDiscriminator(**d_kwargs)
return discriminator, d_kwargs
def build_img_discriminator(args, vocab):
discriminator = None
d_kwargs = {}
d_weight = args.discriminator_loss_weight
d_img_weight = args.d_img_weight
if d_weight == 0 or d_img_weight == 0:
return discriminator, d_kwargs
d_kwargs = {
'arch': args.d_img_arch,
'normalization': args.d_normalization,
'activation': args.d_activation,
'padding': args.d_padding,
}
discriminator = PatchDiscriminator(**d_kwargs)
return discriminator, d_kwargs
