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

1. 训练前准备

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

2. 训练自己的数据集

1. 将下载下来的DCGAN源码解压，如下图所示
   
   代码是完整的，直接运行就可以，无需修改任何语句。

2. 如果是训练自己的数据集，将自己的图像全部调整为96*96大小，也可以改为其他尺寸，但是一定要统一大小，并且是jpg格式。修改后，进入data文件夹，创建新文件夹，如下图所示，我创建的文件夹命名为train。将自己的所有图片放入train文件夹中。
   

3. 在电脑程序中输入CMD打开命令窗口，将路径指定到DCGAN代码路径下，如下图所示。
   

4. 输入命令
   python main.py --dataset train --input_height 96 --input_width 96 --output_height 48 --output_width 48 --crop --train --epoch 20

5. 其中input_height 是指输入图片大小，output_heigh是指输出图片大小，epoch是指训练次数，这里写了20当然是不够的，训练次数越多效果越好，当然时间也越长。到此所有训练工作已经结束。在samples文件夹中可以看到所有训练过程的图片。logs文件夹会记录训练日志。