博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
照片美妆---人脸老化
阅读量:5161 次
发布时间:2019-06-13

本文共 1838 字,大约阅读时间需要 6 分钟。

原文:

本文转载自http://www.sohu.com/a/131656765_642762

论文名称《Age Progression/Regression by Conditional Adversarial Autoencoder》2017 CVPR

论文地址:

该代码是对论文《用条件对抗式自动编码器进行人脸老化、退龄》中算法的Tensorflow实现

环境要求

·Python 2.7x

·Scipy

·TensorFlow

数据库

·FGNET

·MORPH

地址:https://ebill.uncw.edu/C20231_ustores/web/product_detail.jsp?PRODUCTID=8

·CACD

地址:http://bcsiriuschen.github.io/CARC/

·UTKFace (可以从Github 或维基百科获得)

准备训练数据你可以使用任何带年龄标签和性别标签的数据库。在该demo中,我们使用了UTF人脸数据库,因为用这种标定并裁剪过的人脸照片更好一些。请保存并解压UTKFace.tar.gz到文件夹data下。

训练

$ python main.py

训练过程在NVIDIA TITAN X (12GB)上进行了测试。在UTK人脸数据库(23,708张图像,大小为128x128x3)上进行50次epoch的训练时间是两个半小时。

在训练过程中,会建立一个新文件夹save,包括四个子文件夹:summary, samples, test,和checkpoint

·samples :保存每个epoch之后重建的人脸。

·test :保存每个epoch之后的测试结果(基于输入人脸生成的不同年龄的人脸)。

·checkpoint :保存模型。

·summary :保存批损失和中间输出。

用以下命令来可视化summary:

$ cd save/summary

$ tensorboard --logdir .

训练之后,可以检查文件夹samples和test来分别可视化重建和测试性能。下图展示了重建(左)和测试(右)的结果。重建结果(左)的第一行是测试样例,他们分别对应的测试结果(右)由上到下,按年龄增长顺序排列。

不同训练epoch和重建损失的对比如下图所示,为了可视化的目的我们对它进行了低通滤波。原始的记录保存在summary文件夹中。

测试$ python main.py --is_train False --testdir your_image_dir

输入命令之后,应该显示出下面的信息:

Building graph ...

TestingMode

Loadingpre-trained model ...

SUCCESS^_^

Done! Results are saved as save/test/test_as_xxx.png

具体来说,测试人脸会进行两次处理,这两次分别将其视作男性和女性。因此,保存的文件会分别命名为test_as_male.png(作为男性测试) 和 test_as_female.png(作为女性测试)。如果想实现更好的结果,需要在更大并更多样化的数据集上进行训练。

训练过程演示第一行显示了不同年龄的输入人脸,其他行显示了每次epoch之后输出人脸的提高。输出人脸由上到下按年龄递增顺序排列。

文件

·FaceAging.py :类文件,建立并初始化模型,并且实现训练和测试的相关事项。

地址:https://github.com/ZZUTK/Face-Aging-CAAE/blob/master/FaceAging.py

·ops.py :由FaceAging.py的函数组成,实现卷积,反卷积,全卷积,leaky ReLU激活函数,下载并保存图像等操作。

地址:https://github.com/ZZUTK/Face-Aging-CAAE/blob/master/ops.py

·main.py :演示 FaceAging.py。

https://github.com/ZZUTK/Face-Aging-CAAE/blob/master/main.py

posted on
2018-03-13 10:57 阅读(
...) 评论(
...)

转载于:https://www.cnblogs.com/lonelyxmas/p/8554870.html

你可能感兴趣的文章
面试时被问到的问题
查看>>
注解小结
查看>>
201421410014蒋佳奇
查看>>
Xcode5和ObjC新特性
查看>>
CSS属性值currentColor
查看>>
Real-Time Rendering 笔记
查看>>
多路复用
查看>>
处理程序“PageHandlerFactory-Integrated”在其模块列表中有一个错误模块“Manag
查看>>
利用SignalR来同步更新Winfrom
查看>>
反射机制
查看>>
CocoaPod
查看>>
BZOJ 1251: 序列终结者 [splay]
查看>>
5G边缘网络虚拟化的利器:vCPE和SD-WAN
查看>>
MATLAB基础入门笔记
查看>>
【UVA】434-Matty's Blocks
查看>>
Android开发技术周报 Issue#80
查看>>
hadoop2.2.0+hive-0.10.0完全分布式安装方法
查看>>
django知识点总结
查看>>
C++ STL stack、queue和vector的使用
查看>>
使用Reporting Services时遇到的小问题
查看>>