Shortcuts

mmedit.models.editors.srgan.modified_vgg

Module Contents

Classes

ModifiedVGG

A modified VGG discriminator with input size 128 x 128.

class mmedit.models.editors.srgan.modified_vgg.ModifiedVGG(in_channels, mid_channels)[源代码]

Bases: mmengine.model.BaseModule

A modified VGG discriminator with input size 128 x 128.

It is used to train SRGAN and ESRGAN.

参数
  • in_channels (int) – Channel number of inputs. Default: 3.

  • mid_channels (int) – Channel number of base intermediate features. Default: 64.

forward(x)[源代码]

Forward function.

参数

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

返回

Forward results.

返回类型

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.