向AI转型的程序员都关注了这个号👇👇👇
人工智能大数据与深度学习 公众号：datayx
代码说明：
-
1，在yolov5的检测基础上，加上关键点回归分支，请先下载yolov5的工程：https://github.com/ultralytics/yolov5
-
2，detect_one.py是单张图片的测试代码， 基于部分wideface训练的模型，稍后在百度云公开。
代码 获取方式：
分享本文到朋友圈
关注微信公众号 datayx 然后回复 人脸 即可获取。
AI项目体验地址 https://loveai.tech
主要修改代码部分：
-
(1)hyp.scatch.yaml中增加关键点loss的超参数（landmark: 0.5）
-
(2)yolo.py中增加了关键点回归的计算
-
(3)face_datasets.py为人脸数据的读取方式，准备数据的格式参考yolov5的格式，在后面增加关键点的坐标（归一化）
-
(4) loss.py中增加关键点回归的loss计算
-
(5) 链接: https://pan.baidu.com/s/1zjPIF2NZ9CGtB2iUCox6hw 密码: j83n
-
(6) 效果图 :
关于口罩人脸的问题：
- 1，增加口罩人脸这个类别，建议不要直接在检测分支中增加类别。
- 2，应该在关键点分支额外增加一个属性分支，接一个二分类，判断有没有戴口罩。
- 3，这样可以减少口罩人脸的误检问题
关于关键点的问题：
-
1，建议可以替换成wingloss训练，可以优化关键点的精准度。
-
2，可以解决关键点的离群点问题
-
3，wideface之中有不少特别小的人脸，如果不处理会对精度有一定的影响。
机器学习算法AI大数据技术
搜索公众号添加： datanlp
长按图片，识别二维码
阅读过本文的人还看了以下文章：
基于40万表格数据集TableBank，用MaskRCNN做表格检测
《深度学习入门：基于Python的理论与实现》高清中文PDF+源码
2019最新《PyTorch自然语言处理》英、中文版PDF+源码
《21个项目玩转深度学习：基于TensorFlow的实践详解》完整版PDF+附书代码
PyTorch深度学习快速实战入门《pytorch-handbook》
【下载】豆瓣评分8.1,《机器学习实战:基于Scikit-Learn和TensorFlow》
李沐大神开源《动手学深度学习》，加州伯克利深度学习（2019春）教材
【Keras】完整实现‘交通标志’分类、‘票据’分类两个项目，让你掌握深度学习图像分类
如何利用全新的决策树集成级联结构gcForest做特征工程并打分？
Machine Learning Yearning 中文翻译稿
斯坦福CS230官方指南：CNN、RNN及使用技巧速查（打印收藏）
中科院Kaggle全球文本匹配竞赛华人第1名团队-深度学习与特征工程
不断更新资源
深度学习、机器学习、数据分析、python
搜索公众号添加： datayx