AI学习者 · 2020年03月05日

libfacedetection的PyTorch训练程序开源

开源人脸检测库libfacedetection的PyTorch训练程序开源,赶紧试试吧~
授权转自微信

我们前几日刚发布了libfacedetection开源人脸检测算法第三版模型,新增加了五点人脸检测功能。详见文章《算法升级!开源极快速CNN人脸检测新增人脸关键点功能

1.jpg

训练V3这个新模型时,没有继续使用Caffe进行训练,而改用了PyTorch。因为训练时会涉及频繁的模型修改,PyTorch在自定义模型方面比Caffe更加方便。

新的PyTorch训练程序参考了中科院自动化所张士峰的GitHub项目FaceBoxes.PyTorch以及Max deGroot的GitHub项目ssd.pytorch,并把代码整理的更简洁。

之所以能够人脸检测和关键点检测同时做,是采用了下面策略:

  • 前100个Epoch仅训练人脸检测;
  • 后400个Epoch人脸检测和关键点检测交替训练。

本项目的亮点是将训练好的模型进行8位定点化,并自动生成libfacedetection库所需的CPP文件;还可以自动将卷积层conv和批归一化层bn进行融合,使之成为一个新的卷积层,这样不需要再进行bn操作,以提升速度。定点化和层融合详见Readme文档和代码。

项目的网址是 https://github.com/ShiqiYu/libfacedetection.train ,欢迎测试和使用。

-End-

推荐阅读


欢迎关注我的极术专栏:AI搬运小能手,给您分享最前沿靠谱的高质量AI技术干货。
推荐阅读
关注数
3
内容数
8
目录
极术微信服务号
关注极术微信号
实时接收点赞提醒和评论通知
安谋科技学堂公众号
关注安谋科技学堂
实时获取安谋科技及 Arm 教学资源
安谋科技招聘公众号
关注安谋科技招聘
实时获取安谋科技中国职位信息