当前位置：首页 > news >正文

tensorflow2从入门到精通——DCGAN算法实现

news 来源：原创 2024/5/19 20:02:28

class DCGAN(tf.keras.Model):
    def __init__(self):
        super(DCGAN,self).__init__()
        self.Generator = tf.keras.models.Sequential([   #生成器网络
            layers.Dense(7*7*256),
            layers.BatchNormalization(),
            layers.LeakyReLU(),
            layers.Reshape((7,7,256)),
            layers.Conv2DTranspose(128, (5, 5), strides=(1, 1), padding='same'),
            layers.BatchNormalization(),
            layers.LeakyReLU(),
            layers.Conv2DTranspose(64, (5, 5), strides=(2, 2), padding='same'),
            layers.BatchNormalization(),
            layers.LeakyReLU(),
            layers.Conv2DTranspose(1, (5, 5), strides=(2, 2), padding='same',activation='tanh'),
        ])
        self.Discriminator = tf.keras.models.Sequential([   #判别器网络
            layers.Conv2D(64,(5,5),strides=(2,2),padding='same'),
            layers.LeakyReLU(),
            layers.Conv2D(128, (5, 5), strides=(2, 2), padding='same'),
            layers.LeakyReLU(),
            layers.Flatten(),
            layers.Dense(1)
        ])