Chinese Journal of Liquid Crystals and Displays, Volume. 38, Issue 11, 1455(2023)

Infrared dim-small target detection under complex background based on attention mechanism

Ying LIU1,2, Hai-jiang SUN1、*, and Yong-xian ZHAO1,2
Author Affiliations
  • 1Changchun Institute of Optics,Fine Mechanics and Physics,Chinese Academy of Sciences,Changchun 130033,China
  • 2University of Chinese Academy of Sciences,Beijing 100049,China
  • show less

    The small target in infrared images has fewer pixels and lack of obvious feature details in complex scenes, which make it difficult to extract target features and usually has low detection accuracy. This paper proposes an infrared small target detection method based on attention mechanism under complex background. Based on YOLOv5 (You Only Look Once) network, SimAMC3 attention mechanism module is designed to optimize the feature extraction layer of the network. The target detection head is designed by adding a feature fusion layer to change the depth of feature extraction, a new weak target detection layer can be obtained, so that the shallow feature layer can better retain the spatial information of the weak target. Finally, the screening method of prediction box is improved to increase the detection accuracy of objects with close distance or overlapping. In the experiment, two SIRST infrared dim-small target image datasets are selected to label and train them. The experimental results show that compared with the original YOLOv5 algorithm, the improved algorithm improves the mean average accuracy (mAP) by 4.8% and 7.1%. It can effectively detect infrared dim-small targets under different complex backgrounds, reflecting good robustness and adaptability, so it can be effectively applied to detect infrared dim-small target.

    Tools

    Get Citation

    Copy Citation Text

    Ying LIU, Hai-jiang SUN, Yong-xian ZHAO. Infrared dim-small target detection under complex background based on attention mechanism[J]. Chinese Journal of Liquid Crystals and Displays, 2023, 38(11): 1455

    Download Citation

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

    Category: Research Articles

    Received: Jan. 6, 2023

    Accepted: --

    Published Online: Nov. 29, 2023

    The Author Email: Hai-jiang SUN (sunhaijiang@126.com)

    DOI:10.37188/CJLCD.2023-0030

    Topics