利用yolov5与openpose进行摔倒检测
5星
- 浏览量: 0
- 大小:None
- 文件类型：None
简介：
本项目运用YOLOv5模型实现目标检测，并结合OpenPose姿态估计技术，有效识别和监测人体动作，旨在准确检测摔倒事件，为老年人及需要特殊照顾的人群提供安全保障。
通过 yolov5 和 openpose 实现摔倒检测需要的模型文件可以从网盘下载。运行 runOpenpose.py 只执行了 open pose 功能，可以获取人体的关键点图，这些关键点图用于后续 .jit 模型训练，并会保存在 data/test 文件夹中。在 pose.py 中 draw 方法的最后一部分可以控制保存关键点图的位置。
运行 detect.py 时，程序首先进行 yolo 目标检测，在检测到人后，会在 detect.py 的第169行根据框的宽高比判断是否符合标准（后续可自行修改）。接下来，会利用坐标将人的图片提取出来给 openpose 进行人姿态检测。
在 runOpenpose.py 文件中第 159 行也加了一些限制条件（同样可以进行调整），以适应不同的需求。如果希望检测其他姿势：
1. 收集相关图像并运行 runOpenpose.py 获得人体关键点图。
2. 将获取到的关键点图根据需要分类，分别放入 data/train 和 data/test 文件夹中。
3. 运行 action_detect/train.py 来进行模型训练。
全部评论 (0)
还没有任何评论哟~
