Laser & Optoelectronics Progress, Volume. 60, Issue 14, 1428005(2023)

Semi-Supervised Infrared Image Target Detection Algorithm Based on Key Points

Yixuan Shen, Tao Jin*, and Jun Dan
Author Affiliations
  • College of Information Science and Electronic Engineering, Zhejiang University, Hangzhou 310027, Zhejiang, China
  • show less

    A semi-supervised target detection algorithm for infrared images based on CenterNet and OMix enhancement (IRCC-OMix) is proposed to improve target detection accuracy. The prior information of the anchor frame in the infrared image is difficult to determine. Therefore, CenterNet is used as the backbone model to detect the target in the infrared image through key points. A semi-supervised learning method based on teacher-student-network mutual learning is introduced owing to the high cost of infrared image annotation, and a semi-supervised infrared image target detection (IRCC) model based on CenterNet and consistency is designed. The random erasure enhancement in the IRCC model may lead to the disappearance of small targets in the infrared image, which affects the detection performance of the model. Therefore, an object-based image mixing enhancement method is adopted to improve the detection ability of the algorithm for small targets. The experimental results on the public dataset, FLIR, show that the average precision mean (mAP) of the IRCC model reaches 55.3%, which is 1.9 percentage points higher than that of the training using only labeled data. This indicates that the model can fully utilize unlabeled data and improve its robustness. The mAP of the OMix-enhanced IRCC model is 56.8%, which is 1.5 percentage points higher than that of the cutout-enhanced IRCC model and achieves good detection performance.

    Tools

    Get Citation

    Copy Citation Text

    Yixuan Shen, Tao Jin, Jun Dan. Semi-Supervised Infrared Image Target Detection Algorithm Based on Key Points[J]. Laser & Optoelectronics Progress, 2023, 60(14): 1428005

    Download Citation

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

    Category: Remote Sensing and Sensors

    Received: May. 16, 2022

    Accepted: Aug. 4, 2022

    Published Online: Jul. 17, 2023

    The Author Email: Jin Tao (jint@zju.edu.cn)

    DOI:10.3788/LOP221605

    Topics