【干货】深度人脸识别的 Caffe 实现(附模型开源地址及论文下载)
1 新智元报道 1
来源:github
编译:刘小芹
【重要通知:10月18 日世界人工智能大会现场一律凭活动行二维码入场, 上午主论坛于 08:30 正式开始 ,现场座位紧张,请提前签到入场。另外, 不接受现金购票,分论坛票只有下午可以入场】 2016 世界人工智能大会 活动行浏览已经超过 7 万,还有一些已购票付款的公司和机构未换二维码。活动行在大会现场设有咨询席位,但提前换二维码节省签到时间。请团购注册的公司关注,务必提醒参会的同事们手机接收二维码,如果不清晰可以打印出来,现场在签到处换领大会嘉宾胸卡。
大会地点:国家会议中心( C4 入口)
会议签到:上午主论坛 07:30-08:30,下午分论坛 13:00-14:00
会议时间:上午主论坛 08:30- 12:00 ,下午分论坛 14:00-18:00
【关注世界人工智能大会的好友们,福利来了!】 爱奇艺科技频道提供周二上午主论坛的直播( H5地址,各端都可以看 ): http://www.iqiyi.com/l_19rr7t1kdn.html
无法亲临现场的好友们欢迎收藏!
【新智元导读】 本论文对人脸识别和验证任务提出一种新的损失函数,即中心损失。中心损失和softmax损失联合监督学习的CNN,其对深层学习特征的人脸识别能力大大提高。对几个大型人脸基准的实验已经令人信服地证明了该方法的有效性。
相关论文
-
题目:A Discriminative Feature Learning Approachfor Deep Face Recognition
-
作者:Yandong Wen, Kaipeng Zhang, Zhifeng Li*, YuQiao
-
新智元微信公众号回复1015,可下载论文。
论文摘要
卷积神经网络(CNN)已经广泛应用于计算机视觉领域,显著提高了现有计算机视觉水平。在大多数可用的CNN中,softmax损失函数被用作训练深度模型的监视信号。为了增强深度学习特征的识别能力,本论文为人脸识别任务提出一种新的监视信号,称作中心损失函数(center loss)。具体地说,中心损失函数学习每类数据的深层特征的中心,同时惩罚深层特征和它们对应的类中心间的距离。更重要的是,我们证明了这种 中心损失函数是可训练的,而且在CNN中非常容易优化。 通过Softmax损失函数和中心损失函数的联合监视,我们可以训练足够强大的CNN,得到两个关键学习目标的深层特征以及尽可能高的类间分散性和类内紧密性,这对人脸识别来说非常重要。令人鼓舞的是,我们这种联合监视的CNN在几个重要人脸识别基准上取得了最高的准确率,包括Labeled Faces inthe Wild (LFW),YouTube Faces (YTF),和MegaFace Challenge。尤其是,我们的新方法在小型训练集(少于500000幅图像、少于20000个人)协议下的MegaFace(最大的公共领域基准)上得到了最好结果,相比以前的结果有显著的提高,这为人脸识别和人脸验证任务带来新的发展。
CNN架构
图:人脸识别任务中使用的CNN架构(来自论文,下同)
用LFW和YTF数据集实验
LFW和YTF中的部分人脸图像,绿色框是同个人,红色反之,白色框的人脸用于测试:
实验结果
实验中,model A是单一使用softmax损失函数监视的模型,model B是softmax损失和对比损失联合监视的模型,model C是softmax损失和中心损失联合监视的模型。实验结果显示,model C的性能比model A和modelB性能更好,在LFW和YTF中都能得到更高的准确率。
用MegaFaceChallenge数据集实验
MegaFace中的人脸图像样本:
我们对使用不同方法的模型进行了人脸识别和人脸验证实验,结果如下:
人脸识别任务中不同模型的正确率
人脸验证任务中不同模型的正确率
结果显示,使用 softmax 损失函数和中心损失函数联合监视的 model C 能得到更高的正确率。
总结
本论文中我们对人脸识别和验证任务提出一种新的损失函数,即中心损失。中心损失和softmax损失联合监督学习的CNN,其对深层学习特征的人脸识别能力大大提高。对几个大型人脸基准的实验已经令人信服地证明了该方法的有效性。
开源
这篇论文发表于2016欧洲计算机视觉大会(ECCV 2016),论文作者于10月12日和13日开源了训练模型和提取深层特征demo。
开源地址:https://github.com/ydwen/caffe-face
训练模型
1. 安装Caffe,请按照安装指南(http://caffe.berkeleyvision.org/installation.html)进行,确保在使用我们的代码前已经正确安装caffe。
2. 下载人脸数据集,e.g. CAISA-WebFace,VGG-Face, MS-Celeb-1M, MegaFace.
3. 预处理训练人脸图像,包括检测、对齐等。这里我们强烈推荐MTCNN,它是用于人脸识别和对齐非常高效的开源工具。
4. 创建训练集和验证集的列表,放到face_example /data /
5. 指定train和val的数据源
6. 指定FC6层的目标数量
7. 指定中心损失层目标数量和损失权重
8. 训练模型
提取深层特征
1. 编译及配置matcaffe
2. 在face_example/extractDeepFeature.m中指定相应的路径
3. 在Matlab运行extractDeepFeature.m
豪华嘉宾阵容,共飨 AI 盛宴
7 大院长齐聚新智元智库院长圆桌论坛
2016世界人工智能大会技术论坛特设“新智元智库院长圆桌会议”,重量级研究院院长8仙过海,带来最前沿的AI研究动态、剖析技术发展趋势。
八位人工智能界“高手”现场过招,思想的激烈碰撞会给目前的人工智能发展带来哪些洞见?八位院长对行业的发展会有哪些前瞻揭秘?10月18日下午,2016 世界人工智能大会技术论坛,答案将逐层揭晓。
新智元智库院长圆桌汇集了中国人工智能产学研三界最豪华院长阵容:美团技术学院院长刘江担任主持人,微软亚洲工程院院长刘震、IBM中国研究院副院长邵凌、360人工智能研究院院长颜水成、北京理工大学计算机学院副院长黄华、Intel 中国研究院院长宋继强、新华网融媒体未来研究院院长杨溟联袂出席。
2016世界人工智能大会技术论坛议程
扫描二维码购票:
购票链接:http://www.huodongxing.com/go/aiworld2016
也可点击‘阅读原文’直接购票
票务联系 132·4023·6748