mmagic.models.editors.aotgan.aot_encoder

Module Contents

Classes

AOTEncoder

Encoder used in AOT-GAN model.

class mmagic.models.editors.aotgan.aot_encoder.AOTEncoder(in_channels=4, mid_channels=64, out_channels=256, act_cfg=dict(type='ReLU'))

Bases: mmengine.model.BaseModule

Encoder 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: 4.

• mid_channels (int, optional) – Channel number of middle feature. Default: 64.

• out_channels (int, optional) – Channel number of output feature. Default: 256.

• act_cfg (dict, optional) – Config dict for activation layer, “relu” by default.

forward(x)

Forward Function.

Parameters

x (Tensor) – Input tensor with shape of (n, c, h, w).

Returns

Output tensor with shape of (n, c, h’, w’).

Return type

Tensor