niftynet.layer.gan_blocks module

class niftynet.layer.gan_blocks.BaseDiscriminator(name='discriminator', *args, **kwargs)

Bases: niftynet.layer.base_layer.TrainableLayer

class niftynet.layer.gan_blocks.BaseGenerator(name='generator', *args, **kwargs)

Bases: niftynet.layer.base_layer.TrainableLayer

class niftynet.layer.gan_blocks.GANImageBlock(generator, discriminator, clip=None, name='GAN_image_block')

Bases: niftynet.layer.base_layer.TrainableLayer

layer_op(random_source, training_image, conditioning, is_training)