基于上下文语义联合 YOLOv7 的分心驾驶检测算法
DOI:
CSTR:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

TP391

基金项目:

国家自然科学基金青年科学基金(42305158)项目资助


Distracted driving detection algorithm based on contextual semantics combined with YOLOv7
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对分心驾驶检测方法存在实时性差、精度低、可部署性差的问题,提出了一种基于上下文语义增强联合 YOLOv7 的 分心驾驶检测算法。首先将模型 backbone 和 head 部分的 ELAN 模块替换成语义上下文增强模块(contextual transformer, CoT), 提高上下文语义信息的捕获能力。其次,将语义关联增强机制(triplet attention)融入卷积块中,插入 backbone 和 head 的连接头之间以及融合 MP2 模块,强化目标间的关联关系以及提升目标特征提取能力。最后,将自注意力双向 Transformer 模块(Biformer) 模块融合 SPPCSPC 模块,提升模型对分心驾驶中的复杂场景和遮挡目标的处理能力。改进的 YOLOv7 算法 在分心驾驶数据集下平均精度均值(mean average precision,mAP)达到了87.3%,比原算法提高了4.3%,模型参数量减少了 4.7%,每秒传输帧数达到了90 fps,具有较好的检测精度与速度。

    Abstract:

    Aiming at the problems of poor real-time,low precision and poor deployable in distracted driving detection methods,a distracted driving detection algorithm based on contextual semantic enhancement combined with YOLOv7 was proposed.Firstly,ELAN modules in backbone and head of the model are replaced with contextual transformer (CoT)block to improve the ability to capture contextual semantic information.Secondly,the Triplet Attention mechanism is integrated into the convolutional block,inserted between the connectors of backbone and head,and the MP2 module is fused to strengthen the correlation between targets and improve the capability of target feature extraction.Finally,the self-attention bidirectional transformer(Biformer)module is integrated with the SPPCSPC module to improve the model's processing ability for complex scenes and occlusive targets in distracted driving.The mean average precision(mAP)of the improved YOLOv7 algorithm in the distracted driving data set reaches 87.3%, which is 4.3%higher than that of the original algorithm,and the number of model parameters is reduced by 4.7%.The number of frames per second reached 90 fps,with good detection accuracy and speed.

    参考文献
    相似文献
    引证文献
引用本文

李 富,徐 凯,朱灵龙,沈昊君,王 泉.基于上下文语义联合 YOLOv7 的分心驾驶检测算法[J].国外电子测量技术,2024,43(3):121-128

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-06-12
  • 出版日期:
文章二维码