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.

参数

• encoder (dict) – Config of the encoder.

• decoder (dict) – Config of the decoder.

forward(x, mask_in)

Forward Function.

参数

• 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).

返回

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

返回类型

torch.Tensor