训练DCGAN(pytorch官网版本)

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

输出如下

显存占用如下

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

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

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

猜你喜欢

转载自www.cnblogs.com/wzyuan/p/10155471.html