`mmagic.models.editors.stylegan1.stylegan_utils`¶

Module Contents¶

`get_mean_latent`(generator[, num_samples, bs_per_repeat])	Get mean latent of W space in Style-based GANs.
`style_mixing`(generator, n_source, n_target[, ...])

mmagic.models.editors.stylegan1.stylegan_utils.get_mean_latent(generator, num_samples=4096, bs_per_repeat=1024)[source]¶

Get mean latent of W space in Style-based GANs.

Parameters

generator (BaseModule) – Generator of a Style-based GAN.
num_samples (int, optional) – Number of sample times. Defaults to 4096.
bs_per_repeat (int, optional) – Batch size of noises per sample. Defaults to 1024.

Returns

Mean latent of this generator.

Return type

Tensor

mmagic.models.editors.stylegan1.stylegan_utils.style_mixing(generator, n_source, n_target, inject_index=1, truncation_latent=None, truncation=0.7, style_channels=512, **kwargs)[source]¶