mmagic.models.editors.stylegan3.stylegan3
¶
Module Contents¶
Attributes¶
- class mmagic.models.editors.stylegan3.stylegan3.StyleGAN3(generator: ModelType, discriminator: Optional[ModelType] = None, data_preprocessor: Optional[Union[dict, mmengine.Config]] = None, generator_steps: int = 1, discriminator_steps: int = 1, forward_kwargs: Optional[Dict] = None, ema_config: Optional[Dict] = None, loss_config=dict())[源代码]¶
Bases:
mmagic.models.editors.stylegan2.StyleGAN2
Implementation of Alias-Free Generative Adversarial Networks. # noqa.
Paper link: https://nvlabs-fi-cdn.nvidia.com/stylegan3/stylegan3-paper.pdf # noqa
Detailed architecture can be found in
StyleGAN3Generator
andStyleGAN2Discriminator
- test_step(data: dict) mmagic.utils.typing.SampleList [源代码]¶
Gets the generated image of given data. Same as
val_step()
.- 参数
data (dict) – Data sampled from metric specific sampler. More details in Metrics and Evaluator.
- 返回
A list of
DataSample
contain generated results.- 返回类型
SampleList
- val_step(data: dict) mmagic.utils.typing.SampleList [源代码]¶
Gets the generated image of given data. Same as
val_step()
.- 参数
data (dict) – Data sampled from metric specific sampler. More details in Metrics and Evaluator.
- 返回
A list of
DataSample
contain generated results.- 返回类型
SampleList
- train_discriminator(inputs: dict, data_samples: mmagic.structures.DataSample, optimizer_wrapper: mmengine.optim.OptimWrapper) Dict[str, torch.Tensor] [源代码]¶
Train discriminator.
- 参数
inputs (dict) – Inputs from dataloader.
data_samples (DataSample) – Data samples from dataloader.
optim_wrapper (OptimWrapper) – OptimWrapper instance used to update model parameters.
- 返回
A
dict
of tensor for logging.- 返回类型
Dict[str, Tensor]
- train_generator(inputs: dict, data_samples: mmagic.structures.DataSample, optimizer_wrapper: mmengine.optim.OptimWrapper) Dict[str, torch.Tensor] [源代码]¶
Train generator.
- 参数
inputs (dict) – Inputs from dataloader.
data_samples (DataSample) – Data samples from dataloader. Do not used in generator’s training.
optim_wrapper (OptimWrapper) – OptimWrapper instance used to update model parameters.
- 返回
A
dict
of tensor for logging.- 返回类型
Dict[str, Tensor]