Shortcuts

mmedit.models.editors.aotgan.aot_decoder

Module Contents

Classes

AOTDecoder

Decoder used in AOT-GAN model.

class mmedit.models.editors.aotgan.aot_decoder.AOTDecoder(in_channels=256, mid_channels=128, out_channels=3, act_cfg=dict(type='ReLU'))[源代码]

Bases: mmengine.model.BaseModule

Decoder used in AOT-GAN model.

This implementation follows: Aggregated Contextual Transformations for High-Resolution Image Inpainting

参数
  • in_channels (int, optional) – Channel number of input feature. Default: 256.

  • mid_channels (int, optional) – Channel number of middle feature. Default: 128.

  • out_channels (int, optional) – Channel number of output feature. Default 3.

  • act_cfg (dict, optional) – Config dict for activation layer, “relu” by default.

forward(x)[源代码]

Forward Function.

参数

x (Tensor) – Input tensor with shape of (n, c, h, w).

返回

Output tensor with shape of (n, c, h’, w’).

返回类型

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.