mmagic.models.archs.simple_encoder_decoder

Module Contents

Classes

SimpleEncoderDecoder

Simple encoder-decoder model from matting.

class mmagic.models.archs.simple_encoder_decoder.SimpleEncoderDecoder(encoder: dict, decoder: dict, init_cfg: Optional[dict] = None)

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

Forward function.

Returns

The output tensor of the decoder.

Return type

Tensor