前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >yolov5人脸检测,带关键点检测

yolov5人脸检测,带关键点检测

作者头像
机器学习AI算法工程
发布2021-05-11 14:09:12
2.6K0
发布2021-05-11 14:09:12
举报

代码说明:

  • 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之中有不少特别小的人脸,如果不处理会对精度有一定的影响。
本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2021-04-29,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 机器学习AI算法工程 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
相关产品与服务
图像识别
腾讯云图像识别基于深度学习等人工智能技术,提供车辆,物体及场景等检测和识别服务, 已上线产品子功能包含车辆识别,商品识别,宠物识别,文件封识别等,更多功能接口敬请期待。
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档