訓練DCGAN(pytorch官網版本)


將pytorch官網的python代碼當下來,然后下載好celeba數據集(百度網盤),在代碼旁新建celeba文件夾,將解壓后的img_align_celeba文件夾放進去,就可以運行代碼了。

輸出如下

顯存占用如下

與目標檢測模型動不動就顯存不夠比,簡直好太多了。

那最后有一個生成的假的圖像,下面是一個epoch的結果

還有5個epoch的結果,明顯效果更好了

補充下,后來又做了胸片的試驗,因為胸片比較大,所以我resize的尺寸也大一些,變成256x256了,結果,崩了,當然,生成的假圖像也很有規律,那就是長得都一樣。。。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM