mmagic.models.editors.srgan.modified_vgg
¶
Module Contents¶
Classes¶
A modified VGG discriminator with input size 128 x 128. |
- class mmagic.models.editors.srgan.modified_vgg.ModifiedVGG(in_channels, mid_channels)[source]¶
Bases:
mmengine.model.BaseModule
A modified VGG discriminator with input size 128 x 128.
It is used to train SRGAN and ESRGAN.
- Parameters
in_channels (int) – Channel number of inputs. Default: 3.
mid_channels (int) – Channel number of base intermediate features. Default: 64.