Table of Contents
Shortcuts

Source code for 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

[docs]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))