mmagic.models.editors.global_local.gl_disc

Module Contents

Classes

GLDiscs

Discriminators in Global&Local.

class mmagic.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

Parameters

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

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

forward(x)

Forward function.

Parameters

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

Returns

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

Return type

tuple[torch.Tensor]

init_weights()

Init weights for models.