Laser & Optoelectronics Progress, Volume. 60, Issue 24, 2410006(2023)

Infrared Vehicle Detection Algorithm Based on Improved Shuffle-RetinaNet

Xiaochang Fan, Yu Liang, and Wei Zhang*
Author Affiliations
  • School of Microelectronics, Tianjin University, Tianjin 300072
  • show less

    In view of the low detection accuracy and high complexity of current multi-scale vehicle detection algorithms in infrared scenes, an infrared vehicle detection algorithm based on Shuffle-RetinaNet is proposed. On the basis of RetinaNet, the algorithm uses ShuffleNetV2 as the feature extraction network. A dual-branch attention module channel attention module is proposed, which adopts the dual-branch structure and adaptive fusion and enhances the ability to extract the key features of the target in infrared images. To optimize the feature fusion, the algorithm integrates cross-scale connection and fast normalized fusion in some feature layers to enhance the multi-scale feature expression. The calibration factor is set to enhance the task interaction of classification and regression, and the accuracy of target classification and locating is increased. A series of experiments are conducted on a self-built infrared vehicle dataset to verify the effectiveness of the proposed algorithm. The detection accuracy of this algorithm for the self-built vehicle dataset is 92.9%, the number of parameters is 11.74×106, and the number of floating-point operations is 24.35×109. The algorithm exhibits better detection performance on the public dataset FLIR ADAS. Experimental results indicate that the algorithm has advantages in detection accuracy and model complexity, giving it good application value in multi-scale vehicle detection tasks in infrared scenes.

    Tools

    Get Citation

    Copy Citation Text

    Xiaochang Fan, Yu Liang, Wei Zhang. Infrared Vehicle Detection Algorithm Based on Improved Shuffle-RetinaNet[J]. Laser & Optoelectronics Progress, 2023, 60(24): 2410006

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Feb. 27, 2023

    Accepted: Apr. 20, 2023

    Published Online: Dec. 8, 2023

    The Author Email: Zhang Wei (tjuzhangwei@tju.edu.cn)

    DOI:10.3788/LOP230713

    Topics