Shortcuts

mmedit.models.editors.pconv.pconv_encoder_decoder

Module Contents

Classes

PConvEncoderDecoder

Encoder-Decoder with partial conv module.

class mmedit.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

Read the Docs v: latest
Versions
master
latest
stable
zyh-doc-notfound-extend
Downloads
pdf
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.