Real-time detection network for tiny traffic sign using multi-scale attention module

被引:20
|
作者
Yang TingTing [1 ]
Tong Chao [1 ]
机构
[1] Beihang Univ, Sch Comp Sci & Engn, Beijing 100191, Peoples R China
基金
国家重点研发计划; 国家自然科学基金重大研究计划; 北京市自然科学基金; 中国国家自然科学基金;
关键词
tiny object detection; traffic sign detection; multi-scale attention module; real-time; GRADIENTS;
D O I
10.1007/s11431-021-1950-9
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
As one of the key technologies of intelligent vehicles, traffic sign detection is still a challenging task because of the tiny size of its target object. To address the challenge, we present a novel detection network improved from yolo-v3 for the tiny traffic sign with high precision in real-time. First, a visual multi-scale attention module (MSAM), a light-weight yet effective module, is devised to fuse the multi-scale feature maps with channel weights and spatial masks. It increases the representation power of the network by emphasizing useful features and suppressing unnecessary ones. Second, we exploit effectively fine-grained features about tiny objects from the shallower layers through modifying backbone Darknet-53 and adding one prediction head to yolo-v3. Finally, a receptive field block is added into the neck of the network to broaden the receptive field. Experiments prove the effectiveness of our network in both quantitative and qualitative aspects. The mAP@0.5 of our network reaches 0.965 and its detection speed is 55.56 FPS for 512 x 512 images on the challenging Tsinghua-Tencent 100k (TT100k) dataset.
引用
收藏
页码:396 / 406
页数:11
相关论文
共 50 条
  • [1] Real-time detection network for tiny traffic sign using multi-scale attention module
    YANG TingTing
    TONG Chao
    Science China(Technological Sciences), 2022, 65 (02) : 396 - 406
  • [2] Real-time detection network for tiny traffic sign using multi-scale attention module
    TingTing Yang
    Chao Tong
    Science China Technological Sciences, 2022, 65 : 396 - 406
  • [3] Traffic Sign Detection Using a Multi-Scale Recurrent Attention Network
    Tian, Yan
    Gelernter, Judith
    Wang, Xun
    Li, Jianyuan
    Yu, Yizhou
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2019, 20 (12) : 4466 - 4475
  • [4] Improved YOLOv5 network for real-time multi-scale traffic sign detection
    Wang, Junfan
    Chen, Yi
    Dong, Zhekang
    Gao, Mingyu
    NEURAL COMPUTING & APPLICATIONS, 2023, 35 (10): : 7853 - 7865
  • [5] Improved YOLOv5 network for real-time multi-scale traffic sign detection
    Junfan Wang
    Yi Chen
    Zhekang Dong
    Mingyu Gao
    Neural Computing and Applications, 2023, 35 : 7853 - 7865
  • [6] Group multi-scale attention pyramid network for traffic sign detection
    Shen, Lili
    You, Liang
    Peng, Bo
    Zhang, Chuhe
    NEUROCOMPUTING, 2021, 452 : 1 - 14
  • [7] Real-Time and Efficient Multi-Scale Traffic Sign Detection Method for Driverless Cars
    Wang, Xuan
    Guo, Jian
    Yi, Jinglei
    Song, Yongchao
    Xu, Jindong
    Yan, Weiqing
    Fu, Xin
    SENSORS, 2022, 22 (18)
  • [8] Multi-scale traffic sign detection model with attention
    Fan, Bei Bei
    Yang, He
    PROCEEDINGS OF THE INSTITUTION OF MECHANICAL ENGINEERS PART D-JOURNAL OF AUTOMOBILE ENGINEERING, 2021, 235 (2-3) : 708 - 720
  • [9] Improved YOLOX-S Real-Time Multi-Scale Traffic Sign Detection Algorithm
    Wang, Nengwen
    Zhang, Tao
    Computer Engineering and Applications, 2023, 59 (21) : 167 - 175
  • [10] Real-Time Traffic Sign Detection using Capsule Network
    Pari, Neelavathy S.
    Mohana, T.
    Akshaya, V
    2019 11TH INTERNATIONAL CONFERENCE ON ADVANCED COMPUTING (ICOAC 2019), 2019, : 193 - 196