Shortcuts

mmedit.models.editors.global_local.gl_disc

Module Contents

Classes

GLDiscs

Discriminators in Global&Local.

class mmedit.models.editors.global_local.gl_disc.GLDiscs(global_disc_cfg, local_disc_cfg)[源代码]

Bases: mmengine.model.BaseModule

Discriminators in Global&Local.

This discriminator contains a local discriminator and a global discriminator as described in the original paper: Globally and locally Consistent Image Completion

参数
  • global_disc_cfg (dict) – Config dict to build global discriminator.

  • local_disc_cfg (dict) – Config dict to build local discriminator.

forward(x)[源代码]

Forward function.

参数

x (tuple[torch.Tensor]) – Contains global image and the local image patch.

返回

Contains the prediction from discriminators in global image and local image patch.

返回类型

tuple[torch.Tensor]

init_weights()[源代码]

Init weights for models.

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.