前言
看过GAN概念的应该了解生成对抗网络的概念是比较复杂的,对于我这种工科生来说我觉得只要能用就行,所以我主要还是寻找能用的“工具”,直接加上我想要的数据进行处理就好。所以我找到了一篇制作demo的文章:https://zhuanlan.zhihu.com/p/24767059。里面也简单的讲解了一下GAN的原理知识,有兴趣的可以去看看。
DCGAN in Tensorflow
这里用的是Tensorflow和python和python的OpenCV,主要使用的是github大佬https://github.com/carpedm20/DCGAN-tensorflow的代码。
首先确保你满足了大佬的Prerequisites
然后你需要一个大容量的图片库,我是在这个网站上下载的:
http://host.robots.ox.ac.uk/pascal/VOC/voc2012/index.html
这个网站也推荐了很多其它的下载地址:
https://www.pianshen.com/article/2576671659/
在下载的data文件夹中再新建一个picture文件夹,运行指令(参数含义:指定生成的图片的尺寸为48x48,我们图片的大小是64x64,跑300个epoch):
activate tensorflow
或activate tensorflow-gpu
python main.py --input_height 64 --input_width 64 --output_height 48 --output_width 48 --dataset picture --crop --train --epoch 300 --input_fname_pattern “*.jpg”