Infrared and Laser Engineering, Volume. 51, Issue 3, 20210106(2022)

Infrared dim and small target detection based on YOLO-IDSTD algorithm

Xinhao Jiang, Wei Cai, Zhiyong Yang, Peiwei Xu, and Bo Jiang
Author Affiliations
  • Armament Launch Theory and Technology Key Discipline Laboratory of PRC, Rocket Force University of Engineering, Xi′an 710025, China
  • show less

    Aiming at the problem that it is difficult to detect infrared dim and small target accurately and quickly in complex background, a lightweight real-time network model YOLO-IDSTD for infrared dim and small target detection was proposed. Firstly, in order to improve the detection speed, the network structure of the feature extraction part was redesigned, and the Focus module was used to reduce the reasoning time after the input layer. Secondly, in order to enhance the detection ability, the path aggregation network was adopted in the feature fusion part and an improved receptive field block was added. Finally, four-scales detection was increased in the target detection part. Compared with the classical lightweight model YOLOv3-tiny on the infrared dim and small target data set, the recall is increased by 7.57%, the average pricision is increased by 1.92%, and the CPU reasoning speed is increased by 36.1%. The model can balance accuracy and speed, and the amount of calculation and parameters are significantly reduced. The size of the model is compressed to 7.27 MB, which reduces the dependence on the computing power of the hardware platform and realizes the accurate and fast detection of infrared dim and small targets.

    Tools

    Get Citation

    Copy Citation Text

    Xinhao Jiang, Wei Cai, Zhiyong Yang, Peiwei Xu, Bo Jiang. Infrared dim and small target detection based on YOLO-IDSTD algorithm[J]. Infrared and Laser Engineering, 2022, 51(3): 20210106

    Download Citation

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

    Category: Image processing

    Received: Feb. 20, 2021

    Accepted: --

    Published Online: Apr. 8, 2022

    The Author Email:

    DOI:10.3788/IRLA20210106

    Topics