base_dcgan_128x128 源代码

# Copyright (c) OpenMMLab. All rights reserved.
from mmagic.models import DataPreprocessor
from mmagic.models.editors import DCGAN
from mmagic.models.editors.dcgan import DCGANDiscriminator, DCGANGenerator

# define GAN model
[文档]model = dict(
    type=DCGAN,
    noise_size=100,
    data_preprocessor=dict(type=DataPreprocessor),
    generator=dict(type=DCGANGenerator, output_scale=128, base_channels=1024),
    discriminator=dict(
        type=DCGANDiscriminator,
        input_scale=128,
        output_scale=4,
        out_channels=100))