Shortcuts

mmedit.models.base_archs.simple_encoder_decoder

Module Contents

Classes

SimpleEncoderDecoder

Simple encoder-decoder model from matting.

class mmedit.models.base_archs.simple_encoder_decoder.SimpleEncoderDecoder(encoder: dict, decoder: dict, init_cfg: Optional[dict] = None)[source]

Bases: mmengine.model.BaseModule

Simple encoder-decoder model from matting.

Parameters
  • encoder (dict) – Config of the encoder.

  • decoder (dict) – Config of the decoder.

  • init_cfg (dict, optional) – Initialization config dict.

forward(*args, **kwargs) torch.Tensor[source]

Forward function.

Returns

The output tensor of the decoder.

Return type

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.