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

被引:19
|
作者
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 条
  • [41] Traffic sign detection based on multi-scale feature extraction and cascade feature fusion
    Yongliang Zhang
    Yang Lu
    Wuqiang Zhu
    Xing Wei
    Zhen Wei
    The Journal of Supercomputing, 2023, 79 : 2137 - 2152
  • [42] EDN-YOLO: Multi-scale traffic sign detection method in complex scenes
    Han, Yanjiang
    Wang, Fengping
    Wang, Wei
    Zhang, Xin
    Li, Xiangyu
    DIGITAL SIGNAL PROCESSING, 2024, 153
  • [43] Real-Time Traffic Sign Detection via Color Probability Model and Integral Channel Features
    Yang, Yi
    Wu, Fuchao
    PATTERN RECOGNITION (CCPR 2014), PT II, 2014, 484 : 545 - 554
  • [44] A deep residual shrinkage network based on multi-scale attention module for subsea Christmas tree valve leakage detection
    Liu, Guijie
    Chen, Yunqing
    Zhang, Xiulong
    Jiang, Weixiong
    Xie, Yingchun
    Tian, Xiaojie
    Leng, Dingxin
    Li, Zhixiong
    MEASUREMENT, 2022, 198
  • [45] Real-Time Traffic Sign Detection Based on Yolov5-MGC
    Zhu Ningke
    Ge Qing
    Wang Hanwen
    Yu Pengfei
    LASER & OPTOELECTRONICS PROGRESS, 2024, 61 (12)
  • [46] A lightweight multi-scale feature integration network for real-time single image super-resolution
    He, Zheng
    Liu, Kai
    Liu, Zitao
    Dou, Qingyu
    Yang, Xiaomin
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2021, 18 (04) : 1221 - 1234
  • [47] A real-time traffic sign detection in intelligent transportation system using YOLOv8-based deep learning approach
    Tang, Mingdeng
    SIGNAL IMAGE AND VIDEO PROCESSING, 2024, 18 (8-9) : 6103 - 6113
  • [48] A lightweight multi-scale feature integration network for real-time single image super-resolution
    Zheng He
    Kai Liu
    Zitao Liu
    Qingyu Dou
    Xiaomin Yang
    Journal of Real-Time Image Processing, 2021, 18 : 1221 - 1234
  • [49] Tiny SSD: A Tiny Single-shot Detection Deep Convolutional Neural Network for Real-time Embedded Object Detection
    Wong, Alexander
    Shafiee, Mohammad Javad
    Li, Francis
    Chwyl, Brendan
    2018 15TH CONFERENCE ON COMPUTER AND ROBOT VISION (CRV), 2018, : 95 - 101
  • [50] CAFFNet: Channel Attention and Feature Fusion Network for Multi-target Traffic Sign Detection
    Liu, Feng
    Qian, Yurong
    Li, Hua
    Wang, Yongqiang
    Zhang, Hao
    INTERNATIONAL JOURNAL OF PATTERN RECOGNITION AND ARTIFICIAL INTELLIGENCE, 2021, 35 (07)