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文件夾會記錄訓練日誌。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章