| Citation: |
ZHOU Huaping, WU Jin, LI Jingzhao, WU Tao. YOLO Infrared Target Detection Algorithm Combining Parallel Pooling and Self-Distillation[J]. Infrared Technology , 2024, 46(8): 883-891.
|
This study addresses low accuracy in infrared object detection, which is often caused by low contrast and scale discrepancies. A parallel-pooling and self-distillation-YOLO (PPSD-YOLO) algorithm that incorporates several crucial features was proposed to solve this challenge. One of the main contributions of this study is the Fusion-P parallel pooling module, which smoothes the surrounding pixels and avoids overlooking important details. In addition, a small infrared object detection layer was added to enhance the detection accuracy of such objects. The initialization anchor frame for this layer was optimized using the K-means++ algorithm. The proposed algorithm includes a multiscale feature perception module (SA-RFE) in the neck layer, which fuses contextual information from various scales of the target for more accurate detection. During the training process, a modified self-distillation framework was used to rectify misdetected targets in the teacher model, leading to improved detection accuracy in the student model. Tests were conducted using the FLIR dataset to evaluate the proposed algorithm. The results show that PPSD-YOLO outperformed YOLOv7 by 2.7% in terms of mAP. This improvement can be attributed to the incorporation of a parallel pooling module, small-object detection layer, SA-RFE module, and a self-distillation framework. This study presents a comprehensive solution to the challenge of low detection accuracy in infrared object detection. The proposed PPSD-YOLO algorithm integrates advanced features that enhance accuracy and improve the overall performance of the detection system. These findings will be useful for researchers and practitioners in computer vision.
| [1] |
JIAO L, ZHANG F, LIU F, et al. A survey of deep learning-based object detection[J]. IEEE Access, 2019, 7: 128837-128868. DOI: 10.1109/ACCESS.2019.2939201
|
| [2] |
ZOU Z, CHEN K, SHI Z, et al. Object detection in 20 years: a survey[J]. Proceedings of the IEEE, 2023, 111(3): 257-276. DOI: 10.1109/JPROC.2023.3238524
|
| [3] |
KOU R, WANG C, PENG Z, et al. Infrared small target segmentation networks: a survey[J]. Pattern Recognition, 2023, 143: 109788. DOI: 10.1016/j.patcog.2023.109788
|
| [4] |
ZHANG M, ZHANG R, YANGY, et al. ISNet: shape matters for infrared small target detection[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2022: 877-886.
|
| [5] |
DAI Y, WU Y, ZHOU F, et al. Attentional local contrast networks for infrared small target detection[J]. IEEE Transactions on Geoscience and Remote Sensing, 2021, 59(11): 9813-9824. DOI: 10.1109/TGRS.2020.3044958
|
| [6] |
LI S, LI Y, LI Y, et al. Yolo-firi: improved yolo v5 for infrared image object detection[J]. IEEE Access, 2021, 9: 141861-141875. DOI: 10.1109/ACCESS.2021.3120870
|
| [7] |
BAI Y, LI R, GOU S, et al. Cross-connected bidirectional pyramid network for infrared small-dim target detection[J]. IEEE Geoscience and Remote Sensing Letters, 2022, 19: 1-5.
|
| [8] |
LIU Y, PENG Z. Infrared small target detection based on resampling-guided image model[J]. IEEE Geoscience and Remote Sensing Letters, 2021, 19: 1-5.
|
| [9] |
CHEN Y, ZHANG G, MA Y, et al. Small infrared target detection based on fast adaptive masking and scaling with iterative segmentation[J]. IEEE Geoscience and Remote Sensing Letters, 2021, 19: 1-5.
|
| [10] |
WANG C Y, Bochkovskiy A, LIAO H Y M. YOLO v7: trainable bag-of-freebies sets new state-of-the-art for real-time object detectors[C] //Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2023: 7464-7475.
|
| [11] |
LI Y, CHEN Y, WANG N, et al. Scale-aware trident networks for object detection[C]//Proceedings of the IEEE/CVF International Conference on Computer Vision, 2019: 6054-6063.
|
| [12] |
JI M, SHIN S, HWANG S, et al. Refine myself by teaching myself: feature refinement via self-knowledge distillation[C]//Proceedings of the IEEE/ CVF Conference on Computer Vision and Pattern Recognition, 2021: 10664-10673.
|
| [13] |
GOU J, YU B, Maybank S J, et al. Knowledge distillation: a survey[J]. International Journal of Computer Vision, 2021, 129: 1789-1819. DOI: 10.1007/s11263-021-01453-z
|
| [14] |
Hinton G, Vinyals O, Dean J. Distilling the knowledge in a neural network[J]. arXiv preprint arXiv: 1503.02531, 2015.
|
| [15] |
Redmon J, Divvala S, Girshick R, et al. You only look once: unified, real-time object detection[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2016: 779-788.
|
| [16] |
Vadidar M, Kariminezhad A, Mayr C, et al. Robust environment perception for automated driving: a unified learning pipeline for visual-infrared object detection[C]//2022 IEEE Intelligent Vehicles Symposium (Ⅳ), 2022: 367-374.
|
| [17] |
Kera S B, Tadepalli A, Ranjani J J. A paced multi-stage block-wise approach for object detection in thermal images[J]. The Visual Computer, 2022: 1-17.
|
| [18] |
CHEN R, LIU S, MU J, et al. Borrow from source models: efficient infrared object detection with limited examples[J]. Applied Sciences, 2022, 12(4): 1896. DOI: 10.3390/app12041896
|
| [19] |
LI Y, LI S, DU H, et al. YOLO-ACN: Focusing on small target and occluded object detection[J]. IEEE Access, 2020, 8: 227288-227303. DOI: 10.1109/ACCESS.2020.3046515
|
| [20] |
GE Z, LIU S, WANG F, et al. Yolox: Exceeding yolo series in 2021[J]. arXiv preprint arXiv: 2107.08430, 2021.
|
| 1. |
陈健，鲁长春. 考虑遮挡的视频图像运动目标激光跟踪定位方法. 现代电子技术. 2025(03): 76-80 .
| |
| 2. |
程鑫，黄鹂，明德烈. 基于核相关滤波的深空目标跟踪算法. 计算机与数字工程. 2024(10): 2967-2971+2983 .
| |
| 3. |
吴捷，段艳艳，马小虎. 基于KL散度与通道选择的热红外目标跟踪算法. 红外技术. 2023(01): 33-39 .
本站查看
| |
| 4. |
吴捷，马小虎. 基于特征融合与通道感知的无人机红外目标跟踪算法. 激光与红外. 2023(04): 626-632 .
| |
| 5. |
吴捷，马小虎. 融合判别性与细粒度特征的抗遮挡红外目标跟踪算法. 红外技术. 2022(11): 1139-1145 .
本站查看
| |
| 6. |
杨擎宇，王永让，李昊，唐善军，李芳，张国华. 一种自适应红外目标尺寸变化的检测跟踪算法. 红外技术. 2022(11): 1176-1185 .
本站查看
|
Supported by: Beijing Renhe Information Technology Co., Ltd. 
DownLoad: