港中文开源基于PyTorch的多任务人脸识别框架

微信扫一扫,分享到朋友圈

港中文开源基于PyTorch的多任务人脸识别框架

加入极市 专业CV交流群,与 6000+来自腾讯,华为,百度,北大,清华,中科院 等名企名校视觉开发者互动交流!更有机会与 李开复老师 等大牛群内互动!

同时提供每月大咖直播分享、真实项目需求对接、干货资讯汇总,行业技术交流。 关注  极市平台  公众号  , 回复  加群, 立刻申请入群~

今天跟大家分享一款新晋开源的出自香港中文大学MMLab实验室的人脸识别库,其最大特点是支持人脸多任务训练,方便使用PyTorch进行人脸识别的训练、评估、特征提取。

代码链接:

https://github.com/XiaohangZhan/face_recognition_framework

该库本来是用于作者XiaoHangZhan在ECCV 2018论文Consensus-Driven Propagation in Massive Unlabeled Data for Face Recognition中的研究

为什么要用多任务方式训练人脸识别?

作者称,不同的数据集含有不同的属性类别定义,我们往往难以对其进行直接合并,而使用多任务学习,是有效利用数据集的一种方式。

该库特征:

  1. 支持学习框架: Multi-task, Single Task

  2. 支持的损失函数: Softmax Loss, ArcFace

  3. 支持的骨干网类型: ResNet, DenseNet, Inception, InceptionResNet, NASNet, VGG

  4. 支持的人脸基准数据集: Megaface (FaceScrub), IJB-A, LFW

  5. 数据增广策略: 翻转, 缩放, 平移

  6. 可以使用Tensorboard进行在线测试与可视化。

该库的使用非常简单,基本只要下载、设置好数据配置,训练、评估、提取特征都只需要一行命令。

Baseline结果:

感谢作者的开源,欢迎大家研究试用。

https://github.com/XiaohangZhan/face_recognition_framework

-完-

*延伸阅读

添加极市小助手微信 (ID : cv-mart) ,备注: 研究方向-姓名-学校/公司-城市 (如:目标检测-小极-北大-深圳),即可申请加入 目标检测、目标跟踪、人脸、工业检测、医学影像、三维&SLAM、图像分割等极市技术交流群 ,更有每月 大咖直播分享、真实项目需求对接、干货资讯汇总,行业技术交流 一起来让思想之光照的更远吧~

△长按添加极市小助手

△长按关注极市平台

觉得有用麻烦给个在看啦~   

Galaxy Note10 为什么会有「莫奈彩」?

上一篇

高并发系统,如何计算并发量和峰值数据?

下一篇

你也可能喜欢

港中文开源基于PyTorch的多任务人脸识别框架

长按储存图像,分享给朋友