利用Python编写了一个疲劳检测系统,结合了YOLOv5和dlib进行人脸标记。系统可以监测驾驶员的打哈欠次数、抽烟、喝水和玩手机等行为。实时展示眨眼次数、眼睛闭合程度、眨眼持续时间、打哈欠次数和嘴巴张开程度。可通过调用摄像头或对视频进行检测。通过运行video.py对视频进行检测,需将待检测视频命名为input.mp4并放置在指定目录下,检测完成后将生成output.mp4文件。运行main.py可调用电脑摄像头进行实时检测。系统运行需在PyCharm和Anaconda环境下配置好Pytorch-GPU,并安装相关依赖包。若无GPU环境,可使用CPU运行,但会有些卡顿。若对环境配置不熟悉,也可付费代运行,详情请联系微信:18256215256。
用户评论