mmagic.models.editors.deepfillv1.deepfill_disc

Module Contents

Classes

DeepFillv1Discriminators

Discriminators used in DeepFillv1 model.

class mmagic.models.editors.deepfillv1.deepfill_disc.DeepFillv1Discriminators(global_disc_cfg, local_disc_cfg)

Bases: mmengine.model.BaseModule

Discriminators used in DeepFillv1 model.

In DeepFillv1 model, the discriminators are independent without any concatenation like Global&Local model. Thus, we call this model DeepFillv1Discriminators. There exist a global discriminator and a local discriminator with global and local input respectively.

The details can be found in: Generative Image Inpainting with Contextual Attention.

Parameters

• global_disc_cfg (dict) – Config dict for global discriminator.

• local_disc_cfg (dict) – Config dict for 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.