將pytorch官網的python代碼當下來,然后下載好celeba數據集(百度網盤),在代碼旁新建celeba文件夾,將解壓后的img_align_celeba文件夾放進去,就可以運行代碼了。
輸出如下
顯存占用如下
與目標檢測模型動不動就顯存不夠比,簡直好太多了。
那最后有一個生成的假的圖像,下面是一個epoch的結果
還有5個epoch的結果,明顯效果更好了
補充下,后來又做了胸片的試驗,因為胸片比較大,所以我resize的尺寸也大一些,變成256x256了,結果,崩了,當然,生成的假圖像也很有規律,那就是長得都一樣。。。