Skip to content

Discriminator

class Discriminator

Implementation of the discriminator network for the adversarial component of the perceptual loss.

Args
  • patch_size: integer, determines input size as (patch_size, patch_size, 3).

  • kernel_size: size of the kernel in the conv blocks.

Attributes
  • model: Keras model.

  • name: name used to identify what discriminator is used during GANs training.

  • model._name: identifies this network as the discriminator network in the compound model built by the trainer class.

  • block_param: dictionary, determines the number of filters and the strides for each conv block.

__init__

def __init__(patch_size, kernel_size)