def discriminator_factory(type):
if type == 'simple':
return gan_simple.discriminator()
elif type == 'normal':
return gan_normal.discriminator()
elif type == 'deep':
return gan_deep.discriminator()
elif type == 'vanilla_gan':
return gan.discriminator()
def discriminator_factory(type):
if type == 'gan':
return gan_cls.discriminator()
elif type == 'wgan':
return wgan_cls.discriminator()
elif type == 'vanilla_gan':
return gan.discriminator()
elif type == 'vanilla_wgan':
return wgan.discriminator()
def discriminator_factory(type):
if type == 'gan':
return gan_cls.discriminator()
elif type == 'wgan':
return wgan_cls.discriminator()
elif type == 'vanilla_gan':
return gan.discriminator()
elif type == 'vanilla_wgan':
return wgan.discriminator()
elif type == 'inverse_gan':
return gan_inverse.discriminator_inverse()
elif type == 'stage2_gan':
return gan_cls.discriminator2()
def discriminator_factory(type, b_size, h, scale_size, num_channels):
if type == 'gan':
return gan_cls.discriminator()
elif type == 'wgan':
return wgan_cls.discriminator()
elif type == 'vanilla_gan':
return gan.discriminator()
elif type == 'vanilla_wgan':
return wgan.discriminator()
elif type == 'began':
return began.Discriminator(b_size, h, scale_size, num_channels)
elif type == 'acgan':
return acgan.discriminator()
elif type == 'segan':
return gan_cls.discriminator()