训练DCGAN(pytorch官网版本)


将pytorch官网的python代码当下来,然后下载好celeba数据集(百度网盘),在代码旁新建celeba文件夹,将解压后的img_align_celeba文件夹放进去,就可以运行代码了。

输出如下

显存占用如下

与目标检测模型动不动就显存不够比,简直好太多了。

那最后有一个生成的假的图像,下面是一个epoch的结果

还有5个epoch的结果,明显效果更好了

补充下,后来又做了胸片的试验,因为胸片比较大,所以我resize的尺寸也大一些,变成256x256了,结果,崩了,当然,生成的假图像也很有规律,那就是长得都一样。。。


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM