将pytorch官网的python代码当下来,然后下载好celeba数据集(百度网盘),在代码旁新建celeba文件夹,将解压后的img_align_celeba文件夹放进去,就可以运行代码了. 输出如下 显存占用如下 与目标检测模型动不动就显存不够比,简直好太多了. 那最后有一个生成的假的图像,下面是一个epoch的结果 还有5个epoch的结果,明显效果更好了 补充下,后来又做了胸片的试验,因为胸片比较大,所以我resize的尺寸也大一些,变成256x256了,结果,崩了,当然,生成的假图像