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
    Figures & Tables(8)
    Schematic of multi-label classification
    Backbone network for multi-label image classification
    CSRA model
    Topological relationships between dataset multiple labels
    Typical test results display
    • Table 1. Partial parameters of the experiment

      View table
      View in Article

      Table 1. Partial parameters of the experiment

      NameParameter
      Epoch300
      Batch_size16
      OptimizerAdam(lr=0.000 1)
      lr_scheduler

      StepLR

      (Step_size=10 000,Gamma=0.5)

    • Table 2. Results of different algorithm

      View table
      View in Article

      Table 2. Results of different algorithm

      ModelCorrect detectionFalse detectionMiss detectionF1RecallPrecision
      ResNet5077.84%4.98%17.18%0.923 60.883 10.978 3
      ResNet10180.58%7.56%11.86%0.936 00.961 60.920 1
      Efficientnet82.99%6.36%10.65%0.942 50.920 70.978 9
      Multi_path82.65%13.75%3.61%0.948 90.949 70.950 2
      ResNeXt10184.54%13.57%1.89%0.948 00.947 90.949 8
      C_train90.38%4.30%5.33%0.962 60.960 40.970 2
      ML_IRI91.24%3.78%4.98%0.963 20.951 90.985 4
    • Table 3. Results of ablation experiment

      View table
      View in Article

      Table 3. Results of ablation experiment

      ModelCorrect detectionFalse detectionMiss detectionF1RecallPrecision
      ResNet5077.84%4.98%17.18%0.923 60.883 10.978 3
      +ASL81.10%4.81%14.09%0.939 60.907 80.981 8
      +CSRA87.97%2.75%9.28%0.951 50.926 30.989 9
      ML_IRI91.24%3.78%4.98%0.963 20.951 90.985 4
    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