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)[源代码]

Bases: mmengine.model.BaseModule

Simple encoder-decoder model from matting.

参数
  • encoder (dict) – Config of the encoder.

  • decoder (dict) – Config of the decoder.

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

forward(*args, **kwargs) torch.Tensor[源代码]

Forward function.

返回

The output tensor of the decoder.

返回类型

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.