基于改进YOLOv5的驾驶员分心驾驶检测

陈仁祥, 胡超超, 胡小林, 杨黎霞, 张军, 何家乐

吉林大学学报(工学版) ›› 2024, Vol. 54 ›› Issue (04) : 959-968. DOI: 10.13229/j.cnki.jdxbgxb.20220638

基于改进YOLOv5的驾驶员分心驾驶检测

  • 陈仁祥, 胡超超, 胡小林, 杨黎霞, 张军, 何家乐
作者信息 +
History +

摘要

针对采用分类方法进行分心驾驶检测存在只能识别有限分心驾驶行为类别以及忽视时间信息的问题,提出了基于改进YOLOv5的驾驶员分心驾驶检测方法。首先,在YOLOv5的基础上引入Ghost模块,采用线性变换代替部分常规卷积进行特征提取以轻量化网络模型,实现快速又准确地检测图像中手机、水杯、驾驶员双眼和头部区域;其次,在获取目标检测结果的基础上,结合头部姿态估计设计逻辑算法并融入YOLOv5中,从认知分心和视觉分心两个角度检测每帧图像中驾驶员是否存在分心驾驶,避免了分类方法受限分心驾驶类别数的问题,再设置适当的时间阈值,从而实现端到端实时的分心驾驶预警;最后,对采集的18名驾驶员的驾驶行为数据集进行对比试验,验证了本文方法的可行性和有效性。

关键词

分心驾驶 / YOLOv5 / 驾驶行为 / 目标检测 / 头部姿态估计

中图分类号

U491 / TP391.41

引用本文

导出引用
陈仁祥, 胡超超, 胡小林, 杨黎霞, 张军, 何家乐. 基于改进YOLOv5的驾驶员分心驾驶检测. 吉林大学学报(工学版). 2024, 54(04): 959-968 https://doi.org/10.13229/j.cnki.jdxbgxb.20220638

基金

国家重点研发计划项目(2018YFB1306601); 国家自然科学基金项目(51975079); 重庆市自然科学基金创新发展联合基金项目(CSTB2023NSCQ-LZX0127); 重庆市研究生联合培养基地项目(JDLHPYJD2021007); 重庆市教育委员会科学技术研究项目(KJZD-M202200701); 重庆交通大学研究生科研创新项目(2022S0045)

评论

Accesses

Citation

Detail

段落导航
相关文章

/