训练DCGAN(pytorch官网版本)

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

输出如下

显存占用如下

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

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

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

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

原文地址:https://www.cnblogs.com/wzyuan/p/10155471.html