参考https://github.com/tkwoo/anogan-keras
程序运行环境为:
1.环境搭建
(1)anaconda中新建环境
conda create -n gan python=3.6
激活环境
conda activate gan
(2)安装需要的软件库:
pip install opencv-python==3.4.0.14
pip install h5py
pip install scikit-learn
pip install PyQt5
pip install tqdm
pip install tensorflow-gpu==1.5.0
pip install keras==2.1.4
pip install matplotlib
(3)下载模型及权重文件
模型:
sudo git clone https://github.com/tkwoo/anogan-keras
权重文件:
https://github.com/yjucho1/anoGAN/tree/master/assets
下载链接里面的generator和discrimination文件,在anogan-keras文件夹下新建weights文件夹,将上述两个权重文件放进去。
在新建一个result文件,用于存放运行结果。
2.运行模型
分别运行下述代码即可进行训练
cd anogan-keras
python main.py --mode train
参考:
paper : https://arxiv.org/abs/1703.05921
AnoGAN(code, keras) : https://github.com/yjucho1/anoGAN
AnoGAN(code, tf) : https://github.com/LeeDoYup/AnoGAN