ESRGANDiscriminator

class deepinv.models.ESRGANDiscriminator(img_size)

Bases: Module

ESRGAN Discriminator.

The ESRGAN discriminator model was originally proposed by Wang et al.[1]. Implementation taken from edongdongchen/EI.

See Imaging inverse problems with adversarial networks for how to use this for adversarial training.

Parameters:

img_size (tuple) – shape of input image

---

References:

[1]

Xintao Wang, Ke Yu, Shixiang Wu, Jinjin Gu, Yihao Liu, Chao Dong, Yu Qiao, and Chen Change Loy. Esrgan: enhanced super-resolution generative adversarial networks. In Proceedings of the European conference on computer vision (ECCV) workshops, 0–0. 2018.

forward(x)

Forward pass of discriminator model.

Parameters:

x (torch.Tensor) – input image