将pytorch官网的python代码当下来,然后下载好celeba数据集(百度网盘),在代码旁新建celeba文件夹,将解压后的img_align_celeba文件夹放进去,就可以运行代码了。
输出如下
显存占用如下
与目标检测模型动不动就显存不够比,简直好太多了。
那最后有一个生成的假的图像,下面是一个epoch的结果
还有5个epoch的结果,明显效果更好了
将pytorch官网的python代码当下来,然后下载好celeba数据集(百度网盘),在代码旁新建celeba文件夹,将解压后的img_align_celeba文件夹放进去,就可以运行代码了。
输出如下
显存占用如下
与目标检测模型动不动就显存不够比,简直好太多了。
那最后有一个生成的假的图像,下面是一个epoch的结果
还有5个epoch的结果,明显效果更好了