mmagic.models.editors.global_local.gl_encoder_decoder

Module Contents

Classes

GLEncoderDecoder

Encoder-Decoder used in Global&Local model.

class mmagic.models.editors.global_local.gl_encoder_decoder.GLEncoderDecoder(encoder=dict(type='GLEncoder'), decoder=dict(type='GLDecoder'), dilation_neck=dict(type='GLDilationNeck'))

Bases: mmengine.model.BaseModule

Encoder-Decoder used in Global&Local model.

This implementation follows: Globally and locally Consistent Image Completion

The architecture of the encoder-decoder is: (conv2d x 6) –> (dilated conv2d x 4) –> (conv2d or deconv2d x 7)

Parameters

• encoder (dict) – Config dict to encoder.

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

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

forward(x)

Forward Function.

Parameters

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

Returns

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

Return type

torch.Tensor