• Optics and Precision Engineering
  • Vol. 30, Issue 20, 2501 (2022)
Chuankai MIAO1, Shuli LOU1,*, Ting LI2, and Huimin CAI2
Author Affiliations
  • 1School of Physics and Electronic Information, Yantai University, Yantai264005, China
  • 2Tianjin Jinhang Institute of Technical Physics, Tianjin300308, China
  • show less
    DOI: 10.37188/OPE.20223020.2501 Cite this Article
    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 Copy Citation Text show less

    Abstract

    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.
    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