mmagic.models.editors.aotgan.aot_encoder_decoder

Module Contents

Classes

AOTEncoderDecoder

Encoder-Decoder used in AOT-GAN model.

class mmagic.models.editors.aotgan.aot_encoder_decoder.AOTEncoderDecoder(encoder=dict(type='AOTEncoder'), decoder=dict(type='AOTDecoder'), dilation_neck=dict(type='AOTBlockNeck'))

Bases: mmagic.models.editors.global_local.GLEncoderDecoder

Encoder-Decoder used in AOT-GAN model.

This implementation follows: Aggregated Contextual Transformations for High-Resolution Image Inpainting The architecture of the encoder-decoder is: (conv2d x 3) –> (dilated conv2d x 8) –> (conv2d or deconv2d x 3).

参数

• encoder (dict) – Config dict to encoder.

• decoder (dict) – Config dict to build decoder.

• dilation_neck (dict) – Config dict to build dilation neck.