mmedit.models.editors.ttsr.ttsr_disc

Module Contents

Classes

TTSRDiscriminator

A discriminator for TTSR.

class mmedit.models.editors.ttsr.ttsr_disc.TTSRDiscriminator(in_channels=3, in_size=160)

Bases: mmengine.model.BaseModule

A discriminator for TTSR.

参数

• in_channels (int) – Channel number of inputs. Default: 3.

• in_size (int) – Size of input image. Default: 160.

forward(x)

Forward function.

参数

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

返回

Forward results.

返回类型

Tensor