DCGAN之download.py

  • 首先須要導入的包中,gzipzipfile用於文件壓縮和解壓縮相關;argparse用於構建命令行參數;requests用於http請求下載網絡文件資源;subprocess用於運行shell命令;tqdm用於進度條顯示;six包用於python2和python3的兼容,好比 from six.moves import urllib 這句就是導入python2.x的urllib庫。
  • 上面的代碼除了原做者加的註釋以外,我也已經加了一部分註釋,意思應該比較好理解了。主要作的事情,就是利用requests庫從網絡上將mnist,lsun以及celebA這三個數據集下載下來,保存在data目錄下。注意mnistcelebA數據集下載下來以後還進行了解壓縮。
  • 上面的三個數據集,mnist是著名的手寫數字數據庫,你們應該都已經很熟悉了;lsun是大型場景理解數據集(large-scale-scene-understanding);celebA是一個開源的人臉數據庫。除了mnist以外,其他兩個數據集體積都較大,celebA大概有20w+的圖像,壓縮文件體積爲1.4G;而lsun有不少個場景不一樣的數據集,若是按照上面的腳本下載,下載的文件爲bedroom數據集,壓縮文件有46G之大,並且其實下載下來的文件解壓後爲mdb(Access數據庫)格式,不是原始圖片格式,不方便處理。因此咱們實際會下載其餘的數據集做爲替代,好比這個room layout estimation(2G)數據。若是使用download.py腳本下載速度較慢的話,能夠自行下載好數據集,而後放在data目錄下便可。

    DCGAN 代碼簡單解讀
    http://www.mamicode.com/info-detail-2312795.htmlhtml

相關文章
相關標籤/搜索