mmagic.models.editors.aotgan.aot_decoder
¶
Module Contents¶
Classes¶
Decoder used in AOT-GAN model. |
- class mmagic.models.editors.aotgan.aot_decoder.AOTDecoder(in_channels=256, mid_channels=128, out_channels=3, act_cfg=dict(type='ReLU'))[source]¶
Bases:
mmengine.model.BaseModule
Decoder used in AOT-GAN model.
This implementation follows: Aggregated Contextual Transformations for High-Resolution Image Inpainting
- Parameters
in_channels (int, optional) – Channel number of input feature. Default: 256.
mid_channels (int, optional) – Channel number of middle feature. Default: 128.
out_channels (int, optional) – Channel number of output feature. Default 3.
act_cfg (dict, optional) – Config dict for activation layer, “relu” by default.