DCGAN图像生成实战(pytorch)

我自己的配置采用cpu完成训练。文中也可以设置GPU,python版本3.9,在自己用conda创建的虚拟环境中完成。

1、任务解读

原任务是生成嘴部特征图片数据,我采用自建的眼底数据,目标是生成具有特征的眼底图片

 1.1 DCGAN工程技巧

  • 在网络深层去除全连接层
  • 使用带步长的卷积代替池化
  • 在生成器的输出层使用Tanh激活,其它层使用ReLu。Tanh的范围在[-1,1]可以保证图像的范围
  • 在判别器的输出层采用sigmoid激活(因为要的是0-1之间的概率),其它层用了LReLu激活。
  • 除了生成器的输出层和判别器的输入层,其他卷积层上都用了Batch Normalization,可以稳定学习,有助于处理初始化不良导致的训练不收敛的问题。

1.2 模型定义

 nc:图片的通道数(nc=3通道)

nz:噪声的维度(nz=100)

ngf:生成器特征图通道数量单位(ngf=64)

 

 nc:图片的通道数(nc=3通道)

ndf:判别器特征图通道数量单位(ngf=64)

二、模型搭建

三、模型训练。四、模型测试见代码

 视频1-DCGAN图像生成项目解读_哔哩哔哩_bilibili 

github代码GitHub - carpedm20/DCGAN-tensorflow: A tensorflow implementation of "Deep Convolutional Generative Adversarial Networks"

csdn:(53条消息) 从零使用GAN(生成对抗网络)进行图像生成_gan图像生成_hyk今天写算法了吗的博客-CSDN博客

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值