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 条
  • [21] Lightweight multi-scale attention-guided network for real-time semantic segmentation
    Hu, Xuegang
    Liu, Yuanjing
    IMAGE AND VISION COMPUTING, 2023, 139
  • [22] Attention-guided multi-scale infrared real-time detection of pedestrian and vehicle
    Zhang Y.
    Ji K.
    He Z.
    Chen G.
    Hongwai yu Jiguang Gongcheng/Infrared and Laser Engineering, 2024, 53 (05):
  • [23] A lightweight network for traffic sign recognition based on multi-scale feature and attention mechanism
    Wei, Wei
    Zhang, Lili
    Yang, Kang
    Li, Jing
    Cui, Ning
    Han, Yucheng
    Zhang, Ning
    Yang, Xudong
    Tan, Hongxin
    Wang, Kai
    HELIYON, 2024, 10 (04)
  • [24] EMDFNet: Efficient Multi-scale and Diverse Feature Network for Traffic Sign Detection
    Li, Pengyu
    Liu, Chenhe
    Li, Tengfei
    Wang, Xinyu
    Zhang, Shihui
    Yu, Dongyang
    ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING-ICANN 2024, PT II, 2024, 15017 : 120 - 136
  • [25] Real-Time Traffic Sign Detection and Recognition using CNN
    Santos, D.
    Silva, F.
    Pereira, D.
    Almeida, L.
    Artero, A.
    Piteri, M.
    de Albuquerque, V
    IEEE LATIN AMERICA TRANSACTIONS, 2020, 18 (03) : 522 - 529
  • [26] Real-Time Traffic Sign Detection Based on Weighted Attention and Model Refinement
    Jia, Zihao
    Sun, Shengkun
    Liu, Guangcan
    NEURAL PROCESSING LETTERS, 2023, 55 (06) : 7511 - 7527
  • [27] Real-Time Traffic Sign Detection Based on Weighted Attention and Model Refinement
    Zihao Jia
    Shengkun Sun
    Guangcan Liu
    Neural Processing Letters, 2023, 55 : 7511 - 7527
  • [28] A real-time detection method of multi-scale traffic signs based on dynamic pruning strategy
    Jiang Q.
    Rui T.
    Dai J.
    Shao F.
    Lu G.
    Wang J.
    Multimedia Tools and Applications, 2023, 82 (21) : 32519 - 32537
  • [29] Traffic sign recognition using optimized YOLO network based on multi-scale feature extraction and attention mechanism
    Yan, Yong
    Zhao, Haoran
    Li, Yunpeng
    Liu, Xuejun
    Wang, Weiwei
    Guo, Jiacheng
    Zhang, Shuo
    Sha, Yun
    Wang, Xuerui
    Jiang, Yinan
    JOURNAL OF ELECTRONIC IMAGING, 2024, 33 (05)
  • [30] LOROD: Fully Convolutional Network for Real-time Multi-scale Object Detection Algorithm
    Hou, Shaoqi
    Li, Chao
    Liu, Xueting
    Zeng, Yuhao
    Du, Wenyi
    Yin, Guangqiang
    2021 IEEE SMARTWORLD, UBIQUITOUS INTELLIGENCE & COMPUTING, ADVANCED & TRUSTED COMPUTING, SCALABLE COMPUTING & COMMUNICATIONS, INTERNET OF PEOPLE, AND SMART CITY INNOVATIONS (SMARTWORLD/SCALCOM/UIC/ATC/IOP/SCI 2021), 2021, : 579 - 584