Shortcuts

mmedit.models.editors.plain.plain_refiner

Module Contents

Classes

PlainRefiner

Simple refiner from Deep Image Matting.

class mmedit.models.editors.plain.plain_refiner.PlainRefiner(conv_channels=64, init_cfg=None)[源代码]

Bases: mmengine.model.BaseModule

Simple refiner from Deep Image Matting.

参数
  • conv_channels (int) – Number of channels produced by the three main convolutional layer. Default: 64.

  • pretrained (str) – Name of pretrained model. Default: None.

init_weights()[源代码]

Init weights for the module.

forward(x, raw_alpha)[源代码]

Forward function.

参数
  • x (Tensor) – The input feature map of refiner.

  • raw_alpha (Tensor) – The raw predicted alpha matte.

返回

The refined alpha matte.

返回类型

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.