利用YOLOV5与dlib实现疲劳检测
利用Python编写的疲劳检测系统,结合YOLOv5和dlib对人脸进行标记。系统可实时监测驾驶员的打哈欠、抽烟、喝水、玩手机等行为,同时显示眨眼次数、眼睛闭合程度、眨眼持续时间、打哈欠次数及嘴巴张开程度。可选用摄像头实时检测或对视频进行分析。通过运行video.py对视频进行检测,只需将待分析的视频命名为input.mp4并放置在指定目录下,检测完成后将生成output.mp4文件。而通过运行main.py,则可调用电脑摄像头进行实时检测。配置PyCharm和Anaconda环境,安装Pytorch-GPU及相关包,并根据解压路径修改代码中的文件路径即可快速搭建运行环境。若无GPU环境,也可使用CPU,但性能会有所下降。如需代运行服务,请联系wx:18256215256。
下载地址
用户评论
#注释完整