深度学习之基于DCGAN实现手写数字生成

该篇文章与上篇文章内容相差不多,但是主要的网络结构不同,上篇文章采用的是GAN网络结构,而这篇文章采用的是DCGAN网络结构。两者的差异在于以下几点:

(1)使用卷积和去卷积代替池化层。
(2)在生成器和判别器中都添加了批量归一化操作。
(3)去掉了全连接层,使用全局池化层替代。
(4)生成器的输出层使用Tanh 激活函数,其他层使用RELU。
(5)判别器的所有层都是用LeakyReLU 激活函数。
其中最本质的一点就是使用了卷积和去卷积

1.导入库

import numpy as np
import glob,imageio,os,PIL
import matplotlib.pyplot as plt
import tensorflow as tf
from tensorflow.keras import layers

2.数据准备

归一化-打乱-batch

(train_images,train_labels),(_,_) = tf.keras.datasets.mnist.load_data()
train_images = train_images.reshape(train_images.shape[0],28,28,1).astype('float32')
train_images = (train_images - 127.5) / 127.5#归一化到[-1,1]之间
batch_size = 256
buffer_size = 60000
datasets = tf.data.Dataset.from_tensor_slices(train_images).shuffle(buffer_size).batch(batch_size)

3.生成器与判别器的构建

生成器的构建中,采用了tf.keras.layers.Conv2DTranspose,也就是去卷积,它的目的是将经过池化层以后缩小的矩阵扩大到一定的大小,比如说从3 * 3 扩大到 5 * 5,如下图所示:
在这里插入图片描述

def Generator_model():
    model = tf.keras.Sequential([
        tf.keras.layers.Dense(7*7*256,use_bias=False,input_shape=(100,)),#输入一个长度为100的随机向量,
        tf.keras.layers.BatchNormalization(),
        tf.keras.layers.LeakyReLU(),
        tf.keras.layers.Reshape((7,7,256)),#调整为形状为(7,7,256)。

        tf.keras.layers.Conv2DTranspose(128,(5,5),strides=(1,1),padding="same",use_bias=False),
        tf.keras.layers.BatchNormalization(),
        tf.keras.layers.LeakyReLU(),

        tf.keras.layers.Conv2DTranspose(64,(5,5),strides=(2,2),padding="same",use_bias=False),
        tf.keras
  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

starlet_kiss

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值