Windows下用DCGAN训练自己的数据集

最新推荐文章于 2023-06-30 16:06:41 发布

花生米生花@

最新推荐文章于 2023-06-30 16:06:41 发布

阅读量5k

点赞数 2

文章标签： DCGAN TensorFlow

本文链接：https://blog.csdn.net/weixin_42795611/article/details/88928093

版权

1. 训练前准备

DCGAN是在TensorFlow框架下训练数据，所以在使用DCGAN前先配置好TensorFlow环境，可以参考这篇博客win10+cuda9.2+TensorFlow安装。
下载DCGAN源码，百度链接：DCGAN源码，提取码：nd61

将下载下来的DCGAN源码解压，如下图所示

代码是完整的，直接运行就可以，无需修改任何语句。
如果是训练自己的数据集，将自己的图像全部调整为96*96大小，也可以改为其他尺寸，但是一定要统一大小，并且是jpg格式。修改后，进入data文件夹，创建新文件夹，如下图所示，我创建的文件夹命名为train。将自己的所有图片放入train文件夹中。
在电脑程序中输入CMD打开命令窗口，将路径指定到DCGAN代码路径下，如下图所示。
输入命令
python main.py --dataset train --input_height 96 --input_width 96 --output_height 48 --output_width 48 --crop --train --epoch 20
其中input_height 是指输入图片大小，output_heigh是指输出图片大小，epoch是指训练次数，这里写了20当然是不够的，训练次数越多效果越好，当然时间也越长。到此所有训练工作已经结束。在samples文件夹中可以看到所有训练过程的图片。logs文件夹会记录训练日志。