融合特征金字塔和通道注意力的轻量车辆检测算法
DOI:
CSTR:
作者:
作者单位:

西安工程大学电子信息学院 西安 710048

作者简介:

通讯作者:

中图分类号:

TP391.41;U463.6;U495

基金项目:

西安市碑林区应用技术研发项目(GX2007)资助


Lightweight Vehicle Detection Network Fusing Feature Pyramid and Channel Attention
Author:
Affiliation:

School of Electronics and Information, Xi'an Polytechnic University, Xi'an 710048, China

Fund Project:

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

    车辆检测是智能交通、无人驾驶等系统得以实现的重要支撑性技术。低精度或低速度的车辆检测器应用受限,因此提出了一种快速准确的车辆检测器。首先,前端特征提取网络VGG16由MobileNetV3_Large替代,减少了参数量和计算量,并增加了对高维特征的提取能力;其次,利用特征金字塔思想构建双向加权融合网络,有效融合不同尺度的特征,获取多维度的车辆特征;最后在特征提取层引入高效通道注意力,重新标定不同特征通道的重要性,进一步提高模型性能。与SSD相比,所提出的模型在KITTI数据集和BDD 100 K数据集上分别将平均精度提高了7.50%和3.50%,并具有实时检测能力(超过40 fps),在检测精度和速度方面有更好的平衡,说明了方法的有效性。

    Abstract:

    Vehicle detection is an important supporting technology for the realization of intelligent transportation, autonomous driving, etc. Poor accuracy or low inference vehicle detectors are limited in application, therefore this paper proposes a fast and accurate vehicle detector. First, the front-end feature extraction network VGG16 is replaced by MobileNetV3_Large, which reduces the number of parameters and computation, and increases the ability to extract high-dimensional features. Next, the feature pyramid idea is used to construct a weighted bi-directional fusion network to obtain multi-dimensional vehicle features; In the end, introducing efficient channel attention in the feature extraction layer to re-calibrate the importance of different feature channels and further improve the model performance. Compared with SSD, our proposed model improves mAP by 7.50% and 3.50% on KITTI dataset and BDD100K dataset, and with real-time inference (more than 40 FPS), it reports a better trade-off in terms of detection accuracy and speed, illustrating the effectiveness of our method.

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

张 奇,陈梦蝶,赵 杰.融合特征金字塔和通道注意力的轻量车辆检测算法[J].国外电子测量技术,2023,42(01):41-48

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