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)[source]

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)[source]

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

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

Free document hosting provided by Read the Docs.