欢迎访问《汽车安全与节能学报》,

汽车安全与节能学报 ›› 2022, Vol. 13 ›› Issue (1): 122-130.DOI: 10.3969/j.issn.1674-8484.2022.01.012

• 智能驾驶与智慧交通 • 上一篇    下一篇

多通路高分辨率网络与注意力机制融合的车辆检测模型

丁飞1,2(), 米冠宇1, 童恩3, 张楠1, 暴建民1, 张登银1,2   

  1. 1.南京邮电大学 江苏省宽带无线通信和物联网重点实验室,南京 210003,中国
    2.南京邮电大学 通信与网络技术国家工程研究中心,江苏 南京 210003,中国
    3.中国移动通信集团江苏有限公司,南京 210029,中国
  • 收稿日期:2021-09-15 修回日期:2021-12-20 出版日期:2022-03-31 发布日期:2022-04-02
  • 作者简介:丁飞(1981—),男(汉族),江苏,副教授。E-mail:dingfei@njupt.edu.cn
  • 基金资助:
    国家自然科学基金(61872423);工业和信息化部产业技术基础公共服务平台项目(2019-00892-3-1);工业和信息化部通信软科学研究项目(2019-R-26);江苏省“六大人才高峰”高层次人才资助项目(DZXX-008)

Multi-channel high-resolution network and attention mechanism fusion for vehicle detection model

DING Fei1,2(), MI Guanyu1, TONG En3, ZHANG Nan1, BAO Jianmin1, ZHANG Dengyin1,2   

  1. 1. Jiangsu Key Laboratory of Broadband Wireless Communication and Internet of Things, Nanjing University of Posts and Telecommunications, Nanjing 210003, China
    2. National Local Joint Engineering Research Center for Communication and Network Technology, Nanjing University of Posts and Telecommunications, Nanjing 210003, China
    3. China Mobile Group Jiangsu Company Ltd., Nanjing 210029, China
  • Received:2021-09-15 Revised:2021-12-20 Online:2022-03-31 Published:2022-04-02

摘要:

为改善道路交通监测和保证智能网联交通系统的安全、可靠与稳定运行,提出了在路侧边缘平台中基于多通路高分辨率网络与注意力机制融合的车辆检测模型MCHRANet。该模型采用多通路的高分辨率网络的结构设计,保留高分辨率特征并保障识别准确率。融入注意力机制的特征融合方法,通过特征连接权重自学习实现多尺度特征的深度融合。各通路网络采用跳跃连接促进跨层特征融合,加速网络收敛,并利用公开数据集对车辆检测性能进行评估并验证。结果表明:所提模型的车辆检测性能优于3个传统模型,改进后的网络识别平均精度均值(mAP)指标接近95%,且对于不同场景下的检测具有良好的鲁棒性。

关键词: 车辆检测, 高分辨率网络, 注意力机制, 特征融合, 跳跃连接

Abstract:

A vehicle detection model named MCHRANet based on the multi-channel high-resolution network and attention mechanism was proposed to improve the safe, reliable, and stable operation of road traffic monitoring and intelligent connected transportation system (ICTS) in the roadside edge platform. The model adopted multi-channel high-resolution network structure design to retain high-resolution features and ensured detection robustness. The feature fusion method incorporating the attention mechanism realized the deep fusion of multi-scale features through the self-learning of feature connection weights. Each channel network adopted jump connections to promote cross-layer feature fusion and accelerate network convergence. A public data set was used to evaluate the vehicle detection performances in different scenarios. The results show that the vehicle detection performance of the proposed model is better than the three traditional models, the improved network recognition mean average precision (AP) index is close to 95%, and it has good robustness for detection in different scenarios.

Key words: vehicle detection, high-resolution network, attention mechanism, feature fusion, jump connection

中图分类号: