代码进行了部分修改,得到V1.0版本
主要不同地方为:
1、疲劳检测中去掉了点头行为的检测,仅保留闭眼检测和打哈欠检测。
2、Yolov5的权重进行了重新训练,增加了训练轮次。
3、前端UI进行了修改,精简了部分功能。
4、增加图像扩增功能,提升测试多样性。
5、除调用摄像头增加对视频转化为帧图像进行检测并将信息整合resut
该项目为人物专注性检测,分为两个检测部分,疲劳检测和分心行为检测。 疲劳检测部分,使用Dlib进行人脸关键点检测,然后通过计算眼睛和嘴巴的开合程度来判断是存在否闭眼或者打哈欠,并使用Perclos模型计算疲劳程度。 分心行为检测部分,使用Yolov5,检测是否存在玩手机、抽烟、喝水这三种行为。
依赖:YoloV5、Dlib、PySide2
main11.py,即可使用本程序,调用摄像头进行疲劳检测
main1.py,检测视频转化为帧图像
十分感谢原作者的支持和帮助,本项目很大部分都基于源项目,项目所使用的数据集也由原作者提供。