CycleGANDiscriminator

class deeplay.models.discriminators.cyclegan.CycleGANDiscriminator(*args, **kwargs)

Bases: ConvolutionalEncoder2d

CycleGAN discriminator.

Parameters

in_channelsint

Number of channels in the input image.

Examples

>>> discriminator = CycleGANDiscriminator(in_channels=3)
>>> discriminator.build()
>>> x = torch.randn(1, 3, 256, 256)
>>> y = discriminator(x)
>>> y.shape

Return values

The forward method returns the processed tensor.