结合YOLO检测和语义分割的驾驶员安全带检测

被引:17
作者
吴天舒 [1 ,2 ]
张志佳 [1 ]
刘云鹏 [2 ]
郭婉妍 [1 ]
王子韬 [1 ]
机构
[1] 沈阳工业大学软件学院
[2] 中国科学院沈阳自动化研究所光电信息技术研究室
关键词
安全带检测; 语义分割; 交通视频监控; 深度学习;
D O I
暂无
中图分类号
TP391.41 []; U491.61 [];
学科分类号
080203 ;
摘要
为了通过交通监控自动检测驾驶员是否佩戴安全带,提出一种结合目标检测与语义分割的驾驶员安全带检测算法.首先通过设计轻量化目标检测算法完成驾驶员区域快速定位;然后利用经过剪枝加速的语义分割模型对驾驶员区域进行分割,得出安全带连通域;最后通过判断安全带连通域面积检测驾驶员是否佩戴安全带.在驾驶员区域定位和安全带检测2个数据集上进行训练和测试,实验结果表明,驾驶员区域定位算法在精准度为99.96%时速度为73帧/s,安全带检测算法在准确率为94.87%时速度为305帧/s;该算法在兼顾速度的同时具有较好的精准度.
引用
收藏
页码:126 / 131
页数:6
相关论文
共 8 条