Optics and Precision Engineering, Volume. 30, Issue 20, 2501(2022)

Multi-label infrared image classification algorithm based on weakly supervised learning

Chuankai MIAO1... Shuli LOU1,*, Ting LI2 and Huimin CAI2 |Show fewer author(s)
Author Affiliations
  • 1School of Physics and Electronic Information, Yantai University, Yantai264005, China
  • 2Tianjin Jinhang Institute of Technical Physics, Tianjin300308, China
  • show less

    Scene perception and classification of FLIR images is a key technology in target recognition and of great significance to infrared reconnaissance and guidance. To resolve the problem of scene perception and classification of FLIR images, this study proposes a multi-label infrared image classification algorithm based on weakly supervised learning. First, a multi-label image classification technique is applied to FLIR images, and the images of multiple scenes are annotated using weakly supervised techniques. Infrared image features are extracted using the ResNet-50 network with a residual structure. Second, a CSRA module is introduced to capture the different spatial regions occupied by different classes. The CSRA module can improve the feature expression performance and realize the inference calculation of topological relationships between multiple labels. Finally, the advanced loss function ASL is introduced to solve the imbalance of the number of positive and negative labels in multi-label classification. The advanced loss limits the contribution of negative samples to the loss function and focuses attention on the positive samples during training. An experiment shows that the algorithm has good adaptability and accuracy, and the accuracy can exceed 90%. The algorithm can be used to perform multi-label classification with high accuracy and adaptability.

    Tools

    Get Citation

    Copy Citation Text

    Chuankai MIAO, Shuli LOU, Ting LI, Huimin CAI. Multi-label infrared image classification algorithm based on weakly supervised learning[J]. Optics and Precision Engineering, 2022, 30(20): 2501

    Download Citation

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

    Category: Information Sciences

    Received: May. 9, 2022

    Accepted: --

    Published Online: Oct. 27, 2022

    The Author Email: LOU Shuli (shulilou@sina.com)

    DOI:10.37188/OPE.20223020.2501

    Topics