mmagic.models.editors.global_local.gl_encoder

Module Contents

Classes

GLEncoder

Encoder used in Global&Local model.

class mmagic.models.editors.global_local.gl_encoder.GLEncoder(norm_cfg=None, act_cfg=dict(type='ReLU'))

Bases: mmengine.model.BaseModule

Encoder used in Global&Local model.

This implementation follows: Globally and locally Consistent Image Completion

Parameters

• norm_cfg (dict) – Config dict to build norm layer.

• act_cfg (dict) – Config dict for activation layer, “relu” by default.

forward(x)

Forward Function.

Parameters

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

Returns

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

Return type

torch.Tensor