/GAN-keras

里面包含许多GAN算法的Keras源码，可以用于训练自己的模型。

Primary LanguagePythonMIT LicenseMIT

GAN生成对抗网络在Keras当中的实现

注意事项

该库上传时间较早，结构不明确且功能有限，本人进行了一定重置，重置后网络结构与功能会相对更好一些。

DCGAN：https://github.com/bubbliiiing/dcgan-keras
SRGAN：https://github.com/bubbliiiing/srgan-keras

目录

  1. 所需环境 Environment
  2. 仓库内容 WhatsIn
  3. 使用方法 Usage
  4. 参考资料 Reference

所需环境

tensorflow-gpu==1.13.1
keras==2.1.5

仓库内容

  • gan
  • dcgan
  • cgan
  • acgan
  • cogan
  • srgan
  • cyclegan-keras
  • cyclegan-pytorch

使用方法

acgan、cgan、gan、dcgan、cogan

这些gan直接运行其中的代码。

cycleGAN

1、下载数据集
这是斑马to黄种马的数据集下载：
https://people.eecs.berkeley.edu/~taesung_park/CycleGAN/datasets/horse2zebra.zip
苹果to橘子数据集下载：
[https://people.eecs.berkeley.edu/~taesung_park/CycleGAN/datasets/apple2orange.zip] (https://people.eecs.berkeley.edu/~taesung_park/CycleGAN/datasets/apple2orange.zip)
画作to照片数据集下载：
https://people.eecs.berkeley.edu/~taesung_park/CycleGAN/datasets/monet2photo.zip
2、将数据集解压后放入到dataset中
3、运行cyclegan.py进行训练

Reference

https://github.com/eriklindernoren/Keras-GAN
https://github.com/eriklindernoren/PyTorch-GAN