1. 訓練前準備
- DCGAN是在TensorFlow框架下訓練數據,所以在使用DCGAN前先配置好TensorFlow環境,可以參考這篇博客win10+cuda9.2+TensorFlow安裝。
- 下載DCGAN源碼,百度鏈接:DCGAN源碼,提取碼:nd61
2. 訓練自己的數據集
-
將下載下來的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文件夾會記錄訓練日誌。