mmagic.models.editors.deepfillv1.deepfill_refiner

Module Contents

Classes

DeepFillRefiner

Refiner used in DeepFill model.

class mmagic.models.editors.deepfillv1.deepfill_refiner.DeepFillRefiner(encoder_attention=dict(type='DeepFillEncoder', encoder_type='stage2_attention'), encoder_conv=dict(type='DeepFillEncoder', encoder_type='stage2_conv'), dilation_neck=dict(type='GLDilationNeck', in_channels=128, act_cfg=dict(type='ELU')), contextual_attention=dict(type='ContextualAttentionNeck', in_channels=128), decoder=dict(type='DeepFillDecoder', in_channels=256))

Bases: mmengine.model.BaseModule

Refiner used in DeepFill model.

This implementation follows: Generative Image Inpainting with Contextual Attention.

Parameters

• encoder_attention (dict) – Config dict for encoder used in branch with contextual attention module.

• encoder_conv (dict) – Config dict for encoder used in branch with just convolutional operation.

• dilation_neck (dict) – Config dict for dilation neck in branch with just convolutional operation.

• contextual_attention (dict) – Config dict for contextual attention neck.

• decoder (dict) – Config dict for decoder used to fuse and decode features.

forward(x, mask)

Forward Function.

Parameters

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

• mask (torch.Tensor) – Input tensor with shape of (n, 1, h, w).

Returns

Output tensor with shape of (n, c, h’, w’).

Return type

torch.Tensor