基于Yolov5的人脸与关键点检测源码：yolov5_face_landmark
5星
- 浏览量: 0
- 大小:None
- 文件类型：None
简介：
yolov5_face_landmark项目是基于YOLOv5框架开发的人脸及面部关键点检测代码库，适用于实时人脸分析和识别系统。
yolov5_face_landmark 是基于 yolov5 的人脸检测模型，并增加了关键点检测功能。以下是实现该功能的步骤：
1. 首先下载 yolov5 工程。
2. `detect_one.py` 文件用于单张图片测试，使用部分 wideface 数据集训练得到的模型。
主要修改内容包括：
(1) 在 `hyp.scratch.yaml` 中添加了关键点损失函数的相关超参数（landmark: 0.5）；
(2) 修改了 `yolo.py` 文件以支持关键点回归计算；
(3) 编写了 `face_datasets.py`，用于读取人脸数据，并在原有 yolov5 格式的基础上增加了归一化后的关键点坐标信息作为训练样本的一部分。
此外，在 `loss.py` 中添加了针对关键点回归任务的损失函数计算方法。
关于口罩佩戴者的人脸检测问题：
1. 建议增加一个表示“戴口罩人脸”的新类别，而不是直接在现有的目标检测分支中修改。
2. 应该考虑在关键点识别部分进行相应调整以适应这一需求。
全部评论 (0)
还没有任何评论哟~
