mmagic.models.editors.pconv.pconv_encoder_decoder

Module Contents

Classes

PConvEncoderDecoder

Encoder-Decoder with partial conv module.

class mmagic.models.editors.pconv.pconv_encoder_decoder.PConvEncoderDecoder(encoder, decoder)

Bases: mmengine.model.BaseModule

Encoder-Decoder with partial conv module.

Parameters

• encoder (dict) – Config of the encoder.

• decoder (dict) – Config of the decoder.

forward(x, mask_in)

Forward Function.

Parameters

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

• mask_in (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