基于注意力机制融合特征的车辆目标检测方法
DOI:
作者:
作者单位:

1.南京信息工程大学;2.无锡学院;3.无锡松炬科技有限公司

作者简介:

通讯作者:

中图分类号:

TP391.4

基金项目:


Vehicle Object Detection Method Based on Attention Mechanism Integrated Features
Author:
Affiliation:

Fund Project:

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

    为了解决道路监控下的车辆目标检测精度低的问题,本文提出一种改进YOLO v7的车辆检测方法。首先引入跨空间学习的高效多尺度注意机制EMA来提高对特征信息的关注;其次将颈部网络中的SPPCSPC模块替换为SPPFCSPC模块,裁剪CBS层,引入EMA注意力机制,以强化对小目标区域的关注,获取更准确的车辆特征;同时,将EMA注意力引入MP模块中,使网络融合更多重要的特征信息;最后,采用MPDIoU损失函数,加快模型收敛速度并提高检测精度。实验结果表明,改进后的YOLO v7检测精度为86.69%,相比原始YOLO v7网络提高了2.83%,可以有效地提升车辆目标检测精度,为道路视频监控等提供保证。

    Abstract:

    Abstract: To address the issue of low vehicle detection accuracy in road surveillance, this paper proposes an improved vehicle detection method based on YOLO v7. Firstly, we introduce the Efficient Multi-Scale Attention Mechanism (EMA) for cross-space learning to enhance attention to feature information. Secondly, we replace the SPPCSPC module in the neck network with the SPPFCSPC module, trim the CBS layer, and introduce the EMA attention mechanism to strengthen attention to small target areas, thereby obtaining more accurate vehicle features. Additionally, we incorporate the EMA attention into the MP module to fuse more important feature information. Finally, employing the MPDIoU loss function accelerates model convergence and enhances detection accuracy. Experimental results show that the improved YOLO v7 achieves a detection accuracy of 86.69%, which is a 2.83% improvement over the original YOLO v7 network. This enhancement effectively boosts the accuracy of vehicle object detection, providing assurance for applications such as road video surveillance.

    参考文献
    相似文献
    引证文献
引用本文
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2024-04-09
  • 最后修改日期:2024-05-31
  • 录用日期:2024-05-31
  • 在线发布日期:
  • 出版日期: