mmagic.models.losses.adv_loss

Module Contents

Classes

ImagePool	Defined a image pool for RelativisticDiscLoss.
DiscLoss	Defined a criterion to calculator loss.
RelativisticDiscLoss	Defined a criterion to calculator loss.
RelativisticDiscLossLS	Defined a criterion to calculator loss.
DiscLossWGANGP	Defined a criterion to calculator loss.
AdvLoss	Returns the loss of discriminator with the specified type for

class mmagic.models.losses.adv_loss.ImagePool(pool_size)

Defined a image pool for RelativisticDiscLoss.

add(images)

query()

class mmagic.models.losses.adv_loss.DiscLoss(gan_type='vanilla')

Bases: torch.nn.Module

Defined a criterion to calculator loss.

name()

return name of criterion.

forward(net, fakeB, realB, model='discriminator')

Get discriminator or generator loss.

class mmagic.models.losses.adv_loss.RelativisticDiscLoss

Bases: torch.nn.Module

Defined a criterion to calculator loss.

name()

return name of criterion.

forward(net, fakeB, realB, model='discriminator')

Get discriminator or generator loss.

class mmagic.models.losses.adv_loss.RelativisticDiscLossLS

Bases: torch.nn.Module

Defined a criterion to calculator loss.

name()

return name of criterion.

forward(net, fakeB, realB, model='discriminator')

Get discriminator or generator loss.

class mmagic.models.losses.adv_loss.DiscLossWGANGP

Bases: DiscLoss

Defined a criterion to calculator loss.

name()

return name of criterion.

forward(net, fakeB, realB, model='discriminator')

Get discriminator or generator loss.

class mmagic.models.losses.adv_loss.AdvLoss

Returns the loss of discriminator with the specified type for DeblurGanv2.

Parameters

loss_type (Str) – One of value in [wgan-gp,lsgan,gan,ragan,ragan-ls].