mmagic.models.archs.patch_disc

Module Contents

Classes

PatchDiscriminator

A PatchGAN discriminator.

class mmagic.models.archs.patch_disc.PatchDiscriminator(in_channels: int, base_channels: int = 64, num_conv: int = 3, norm_cfg: dict = dict(type='BN'), init_cfg: Optional[dict] = dict(type='normal', gain=0.02))

Bases: mmengine.model.BaseModule

A PatchGAN discriminator.

Parameters

• in_channels (int) – Number of channels in input images.

• base_channels (int) – Number of channels at the first conv layer. Default: 64.

• num_conv (int) – Number of stacked intermediate convs (excluding input and output conv). Default: 3.

• norm_cfg (dict) – Config dict to build norm layer. Default: dict(type=’BN’).

• init_cfg (dict) – Config dict for initialization. type: The name of our initialization method. Default: ‘normal’. gain: Scaling factor for normal, xavier and orthogonal. Default: 0.02.

forward(x: torch.Tensor) → torch.Tensor

Forward function.

Parameters

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

Returns

Forward results.

Return type

Tensor

init_weights() → None

Initialize weights for the model.

Parameters

pretrained (str, optional) – Path for pretrained weights. If given None, pretrained weights will not be loaded. Default: None.