成功运行RealnessGAN,并进行数据增强
前言
RealnessGAN网络是《Real or Not Real, that is the Question》这篇文章于2020年发表在ICLR中提出的一个GAN网络结构。本人主要是想使用这个网络进行数据增强。本文主要是想记录本人在运行代码时遇到的问题,方便记忆。
一、环境
Windows10
Anaconda3(python3.6.12)
GPU:GTX2060
cuda10.0
pytorch1.4.0
代码readme推荐的环境是python=3.6、pytorch=1.1.0,但我发现使用pytorch=1.4.0也能跑,所以就没按它的要求来。
二、下载代码
RealnessGAN代码链接:链接: lhttps://github.com/kam1107/RealnessGAN.git.
这个代码是文章作者发布的。文章下载链接: https://openreview.net/forum?id=B1lPaCNtPB.
三、创建文件夹
主要是创建两个文件夹,一个用于存放输入的数据,一个用于保存输出的数据于训练模型。
四、修改代码
1.修改options.py
主要是修改输入输出的路径和一些参数比如输入图像的大小,对