
危险驾驶检测器:运用SSD目标检测算法识别闭眼及接打电话、抽烟等行为,基于PER...
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
简介:本项目研发了一种利用SSD目标检测技术的危险驾驶行为监测系统,能有效识别驾驶员打手机、吸烟和闭眼等高风险动作,并提出相应的预警措施以保障行车安全。
运行环境:1. Python 3.7.42, PyTorch 1.4.0, OpenCV
预训练的权重文件为 vgg_16,具体的配置请参见 Config.py 文件。
- 训练命令: python Train.py
- 单张图片测试: python Test.py
- 视频检测: python camera_detection.py
目前进度:
1. PERCLOS计算完成
2. 眨眼频率计算完成
3. 打哈欠检测及计算完成
4. 疲劳检测完成
5. 人脸情绪检测完成
网络检测性能:准确率82.18%
主要文件说明:
- ssd_net_vgg.py: 定义 SSD 类的文件
- Train.py: 训练代码
- voc0712.py: 数据集处理代码(未修改文件名,需要改的话其他相关代码也要相应改动)
- loss_function.py: 损失函数定义
全部评论 (0)
还没有任何评论哟~


