niftynet.layer.gan_blocks module

class GANImageBlock(generator, discriminator, clip=None, name='GAN_image_block')[source]

Bases: niftynet.layer.base_layer.TrainableLayer

layer_op(random_source, training_image, conditioning, is_training)[source]
class BaseGenerator(name='generator', *args, **kwargs)[source]

Bases: niftynet.layer.base_layer.TrainableLayer

class BaseDiscriminator(name='discriminator', *args, **kwargs)[source]

Bases: niftynet.layer.base_layer.TrainableLayer