yolov5_face_landmark: 基于yolov5的人脸检测与关键点检测

yolov5_face_landmark: 基于yolov5的人脸检测与关键点检测

去发现同类优质开源项目:https://gitcode.com/

简介

yolov5_face_landmark 是一款基于yolov5框架开发的人脸检测及关键点检测工具。在原有yolov5人脸检测功能的基础上，增加了关键点回归的分支，能够同时完成人脸检测与面部关键点的定位。

代码说明

1. 本项目在yolov5的检测基础上，添加了关键点回归分支。请先获取yolov5的基础工程。
2. detect_one.py 是用于测试单张图片的代码，基于部分wideface训练的模型。请注意，该模型将在百度云上公开。
3. 主要的代码修改如下：
   • 在 hyp.scatch.yaml 中增加了关键点loss的超参数（landmark: 0.5）。
   • 在 yolo.py 中增加了关键点回归的计算。
   • face_datasets.py 负责人脸数据的读取方式，数据格式需参考yolov5的格式，并在后面增加关键点的坐标（进行归一化处理）。
   • 在 loss.py 中增加了关键点回归的loss计算。

关于口罩人脸的处理

• 增加了口罩人脸的类别。建议不要直接在检测分支中增加类别。
• 对关键点分支进行相应调整，以适应口罩人脸的特殊情况。

注意事项

• 使用前请确保已下载并配置好yolov5基础工程。
• 代码中的模型文件需自行从百度云获取（获取方式已在描述中提到，但此处不提供具体链接）。
• 测试及使用过程中，请确保所有依赖都已正确安装。

本项目提供的工具和代码旨在帮助开发者快速实现人脸检测与关键点定位的功能，希望能对相关领域的研究与应用有所帮助。

去发现同类优质开源项目:https://gitcode.com/