keras DCGAN

最新推荐文章于 2023-05-14 18:08:31 发布

lsh呵呵

最新推荐文章于 2023-05-14 18:08:31 发布

阅读量3.9k

点赞数 2

分类专栏： python与人工睿智机器学习入门与放弃

≧◔◡◔≦ 如觉得有所帮助，请点个✌赞或关注✌。转载请注明出处。 ≧◔◡◔≦

本文链接：https://blog.csdn.net/nima1994/article/details/83620725

版权

训练结果：
在这里插入图片描述

在这里插入图片描述

代码：
代码基于 eriklindernoren/Keras-GAN ，并修改了trainable与compile 易于混淆的代码。

from keras.datasets import mnist
from keras.layers import Input, Dense, Reshape, Flatten, Dropout
from keras.layers import BatchNormalization, Activation, ZeroPadding2D
from keras.layers.advanced_activations import LeakyReLU
from keras.layers.convolutional import UpSampling2D, Conv2D
from keras.models import Sequential, Model
from keras.optimizers import Adam
import matplotlib.pyplot as plt
import numpy as np


class DCGAN():
    def __init__(self):
        # Input shape
        self.img_rows = 28
        self.img_cols = 28
        self.channels = 1
        self.img_shape = (self.img_rows, self.img_cols, self.channels)
        self.latent_dim = 100

        optimizer = Adam(0.0002, 0.5)

        base_generator = self.build_generator()
        base_discriminator = self.build_discriminator()
        ########
        self.generator = Model(
            inputs=base_generator.inputs,
            outputs=base_generator.outputs)

        self.discriminator = Model(
            inputs=base_discriminator.inputs,
            outputs=base_discriminator.outputs)
        self.discriminator.compile(loss='binary_crossentropy',
                                   optimizer=optimizer,
                                   metrics=['accuracy'])

        frozen_D = Model(
            inputs=base_discriminator.inputs,
            outputs=base_discriminator.outputs)
        frozen_D.trainable = False
        z = Input(shape=(self.latent_dim,))
        img = self.generator(z)
        valid = frozen_D(img)
        self.combined = Model(z, valid