bubbliiiing / GAN-keras

里面包含许多GAN算法的Keras源码,可以用于训练自己的模型。
MIT License
235 stars 78 forks source link

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

仓库内容

使用方法

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