文件名称:train_cnn_GANs:GAN:DCGAN
文件大小:3.54MB
文件格式:ZIP
更新时间:2024-05-20 13:11:14
Python
搭建的一个GAN训练框架,以DCGAN为例。 此框架主要针对GAN生成图片, 后面还会陆续搭建其它框架,欢迎关注。 使用说明: 搭建时使用的环境为:Python3.5, tensorflow1.7 变量设置参考config.py。 详细说明参见config.py。 运行代码: python main.py 另外此代码没有加tensorboard,可以自行加上,然后使用:tensorboard --logdir #_log查看。 其中,ckpt_pb.py:ckpt转pb的代码;GANs_inference.py:生成器生成图片。 训练集:是以前做人脸检测在wider_face上,不同IOU的48x48的图像。生成图中模糊和角度与训练集相关。 下面是迭代10次的效果: 迭代20次的效果图: 之后更新迭代100次的效果图。 对dl感兴趣,还可以关注我的博客,这是我的博客目录:(地址: ) 本
【文件预览】:
train_cnn_GANs-master
----lp.jpg(68KB)
----test_accuracy.py(2KB)
----model()
--------.DS_Store(6KB)
--------README.md(22B)
----pretrain()
--------.DS_Store(6KB)
--------README.md(40B)
----Gan.png(382KB)
----README.md(25KB)
----config.py(1KB)
----test_tfRecords.py(3KB)
----train.txt(62KB)
----GANs_inference.py(9KB)
----lib()
--------loss()
--------optimizer()
--------model()
--------data_load()
--------.DS_Store(10KB)
--------utils()
--------data_aug()
--------train()
----ckpt_pb.py(5KB)
----main.py(2KB)
----g_tfrecords.py(6KB)
----gender()
--------1female()
--------2many()
--------3other()
--------0male()