DAI Jian, ZHAO Xu, LI Lianpeng, LIU Wen, CHU Xinyue. Improved YOLOv5-based Infrared Dim-small Target Detection under Complex Background[J]. Infrared Technology , 2022, 44(5): 504-512.
Citation: DAI Jian, ZHAO Xu, LI Lianpeng, LIU Wen, CHU Xinyue. Improved YOLOv5-based Infrared Dim-small Target Detection under Complex Background[J]. Infrared Technology , 2022, 44(5): 504-512.

Improved YOLOv5-based Infrared Dim-small Target Detection under Complex Background

More Information
  • Received Date: September 17, 2021
  • Revised Date: October 20, 2021
  • Using the traditional algorithm to meet the detection requirements of interference factors, such as complex background and noise, relying on the precise separation and information extraction of infrared targets and environmental background, is difficult. This paper presents a dim–small target detection method for an infrared imaging algorithm based on the improved YOLOv5 for complex backgrounds. Based on YOLOv5, an attention mechanism is introduced in the algorithm to improve the feature extraction ability and detection efficiency. In addition, the loss function and prediction box screening method of the original YOLOv5 target detection network are used to improve the accuracy of the algorithm for infrared dim–small target detection. In the experiment, seven sets of infrared dim–small target image datasets with different complex backgrounds are selected, the data are labeled and trained, and an infrared dim–small target detection model is established. Finally, the accuracy of the algorithm and model is evaluated in terms of the model training and target detection results. The experimental results show that the model trained by employing the improved YOLOv5 algorithm in this study has a significant improvement in detection accuracy and speed compared with several target detection algorithms used in the experiment, and the average accuracy can reach more than 99.6%. The model can effectively detect infrared dim–small targets in different complex backgrounds, and the leakage and false alarm rates are low.
  • [1]
    GAO Jinyan, GUO Yulan, LIN Zaiping, et al. Robust infrared small target detection using multiscale gray and variance difference measures[J]. IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, 2018, 11(12): 5039-5052. DOI: 10.1109/JSTARS.2018.2877501
    [2]
    XU Yongli, WANG Weihua. A method for single frame detection of infrared dim small target in complex background[J]. Journal of Physics. Conference Series, 2020, 1634(1): 012063. DOI: 10.1088/1742-6596/1634/1/012063
    [3]
    马铭阳, 王德江, 孙翯, 等. 基于稳健主成分分析和多点恒虚警的红外弱小目标检测[J]. 光学学报, 2019, 39(8): 0810001. https://www.cnki.com.cn/Article/CJFDTOTAL-GXXB201908013.htm

    MA Mingyang, WANG Dejiang, SUN He, et al. Infrared dim-small target detection based on robust principal component analysis and multi-point constant false alarm[J]. Acta Photonica Sinica, 2019, 39(8): 0810001. https://www.cnki.com.cn/Article/CJFDTOTAL-GXXB201908013.htm
    [4]
    杨其利, 周炳红, 郑伟, 等. 基于全卷积网络的红外弱小目标检测算法[J]. 红外技术, 2021, 43(4): 349-356. http://hwjs.nvir.cn/article/id/3abe1943-0aea-49e7-b39f-cfad3d9c8ac4

    YANG Qili, ZHOU Binghong, ZHENG Wei, et al. Infrared weak target detection algorithm based on full convolutional network[J]. Infrared Technology, 2021, 43(4): 349-356. http://hwjs.nvir.cn/article/id/3abe1943-0aea-49e7-b39f-cfad3d9c8ac4
    [5]
    仇国庆, 杨海静, 王艳涛, 等. 基于视觉特征融合的机载红外弱小目标检测[J]. 激光与光电子学进展, 2020, 57(18): 79-86. https://www.cnki.com.cn/Article/CJFDTOTAL-JGDJ202018007.htm

    QIU Guoqing, YANG Haijing, WANG Yantao, et al. Dim target detection in airborne infrared images based on visual feature fusion[J]. Laser & Optoelectronics Progress, 2020, 57(18): 79-86. https://www.cnki.com.cn/Article/CJFDTOTAL-JGDJ202018007.htm
    [6]
    陆福星, 李夜金, 陈忻, 等. 基于Top-hat变换的PM模型弱小目标检测[J]. 系统工程与电子技术, 2018, 40(7): 1417-1422. https://www.cnki.com.cn/Article/CJFDTOTAL-XTYD201807001.htm

    LU Fuxing, LI Yegin, CHEN Xin, et al. Weak target detection for PM model based on Top-hat transform[J]. Systems Engineering and Electronics, 2018, 40(7): 1417-1422. https://www.cnki.com.cn/Article/CJFDTOTAL-XTYD201807001.htm
    [7]
    汪烈兵, 姜雄飞, 石春光, 等. 基于图像滤波与Hough变换的红外弱小目标检测[J]. 红外技术, 2020, 42(7): 683-687. http://hwjs.nvir.cn/article/id/hwjs202007012

    WANG Liebing, JIANG Xiongfei, SHI Chunguang, et al. Infrared small target detection based on image filtering and Hough transform[J]. Infrared Technology, 2020, 42(7): 683-687. http://hwjs.nvir.cn/article/id/hwjs202007012
    [8]
    韩金辉, 董兴浩, 蒋亚伟, 等. 基于局部对比度机制的红外弱小目标检测算法[J]. 红外技术, 2021, 43(4): 357-366. http://hwjs.nvir.cn/article/id/29b77b73-8c1e-4251-9ae4-c9f39e265270

    HAN Jinhui, DONG Xinjhao, JIANG Yawei, et al. Infrared small dim target detection based on local contrast mechanism[J]. Infrared Technology, 2021, 43(4): 357-366. http://hwjs.nvir.cn/article/id/29b77b73-8c1e-4251-9ae4-c9f39e265270
    [9]
    Kaveh Ahmadi, Ezzatollah Salari. Small dim object tracking using frequency and spatial domain information[J]. Pattern Recognition, 2016, 58: 227-234. DOI: 10.1016/j.patcog.2016.04.001
    [10]
    冯洋. 基于小波包变换的红外弱小目标检测[J]. 计算机与现代化, 2020(12): 112-115, 122. DOI: 10.3969/j.issn.1006-2475.2020.12.019

    FENG Yang. Infrared dim small target detection based on wavelet packet transform[J]. Computers and Modernization, 2020(12): 112-115, 122. DOI: 10.3969/j.issn.1006-2475.2020.12.019
    [11]
    李德新, 钟洪. 基于差异梯度直方图与显著性映射的红外弱小目标检测算法[J]. 光学技术, 2021, 47(5): 594-600. https://www.cnki.com.cn/Article/CJFDTOTAL-GXJS202105015.htm

    LI Dexin, ZHONG Hong. Infrared dim small target detection algorithm based on difference histogram of oriented gradients and saliency map[J]. Optical Technique, 2021, 47(5): 594-600. https://www.cnki.com.cn/Article/CJFDTOTAL-GXJS202105015.htm
    [12]
    ZHANG N, Donahue J, Girshick R, et al. Part-based R-CNNs for fine-grained category detection[C]//European Conference on Computer Vision, 2014: 834-849.
    [13]
    Girshick R. Fast R-CNN[C]//Proceedings of the IEEE International Conference on Computer Vision, 2015: 1440-1448.
    [14]
    REN S, HE K, Girshick R, et al. Faster R-CNN: Towards real-time object detection with region proposal networks[J]. IEEE Transactions on Pattern Analysis & Machine Intelligence, 2017, 39(6): 1137-1149.
    [15]
    Redmon J, Divvala S, Girshick R, et al. You Only Look Once: real-time object detection[C]// IEEE Conference on Computer Vision and Pattern Recognition, 2016: 779-788.
    [16]
    LIU W, Anguelov D, Erhan D, et al. SSD: Single shot multibox detector[C]//European Conference on Computer Vision, 2016: 21-37.
    [17]
    杨晓玲, 江伟欣, 袁浩然. 基于Yolov5的交通标志识别检测[J]. 信息技术与信息化, 2021(4): 28-30. DOI: 10.3969/j.issn.1672-9528.2021.04.005

    YANG Xiaoling, JIANG Weixin, YUAN Haoran. Traffic sign recognition detection based on Yolov5[J]. Information Technology and Informatization, 2021(4): 28-30. DOI: 10.3969/j.issn.1672-9528.2021.04.005
    [18]
    Lowe D G. Distinctive image features from scale invariant keypoints[J]. International Journal of Computer Vision, 2004, 60(2): 91-110. DOI: 10.1023/B:VISI.0000029664.99615.94
    [19]
    范有臣, 马旭, 马淑丽, 等. 深度学习在激光干扰效果评估中的应用[J/OL]. 红外与激光工程, [2021-08-03]: 1-8.

    FAN Youchen, MA Xu, MA Shuli et al. Evaluation method of laser jamming learning effect based on deep learning[J/OL]. Infrared and Laser Engineering, [2021-08-03]: 1-8.
    [20]
    林清平, 张麒麟, 肖蕾. 采用改进YOLOv5网络的遥感图像目标识别方法[J]. 空军预警学院学报, 2021, 35(2): 117-120. https://www.cnki.com.cn/Article/CJFDTOTAL-KLDX202102008.htm

    LIN Qingping, ZHANG Qilin, XIAO Lei. A remote sensing image target recognition method using improved YOLOv5 network[J]. Journal of Air Force Early Warning Academy, 2021, 35(2): 117-120. https://www.cnki.com.cn/Article/CJFDTOTAL-KLDX202102008.htm
    [21]
    王宇宁, 庞智恒, 袁德明. 基于YOLO算法的车辆实时检测[J]. 武汉理工大学学报, 2016, 38(10): 41-46. DOI: 10.3963/j.issn.1671-4431.2016.10.008

    WANG Yuning, PANG Zhiheng, YUAN Deming. Vehicle detection based on YOLO in real time[J]. Journal of Wuhan University of Technology, 2016, 38(10): 41-46. DOI: 10.3963/j.issn.1671-4431.2016.10.008
    [22]
    周飞燕, 金林鹏, 董军. 卷积神经网络研究综述[J]. 计算机学报, 2017, 40(6): 1229-1251. https://www.cnki.com.cn/Article/CJFDTOTAL-JSJX201706001.htm

    ZHOU Feiyan, JIN Linpeng, DONG Jun. Review of convolutional neural network[J]. Chinese Journal of Computers, 2017, 40(6): 1229-1251. https://www.cnki.com.cn/Article/CJFDTOTAL-JSJX201706001.htm
    [23]
    ZHOU Junchi et al. Ship target detection algorithm based on improved YOLOv5[J]. Journal of Marine Science and Engineering, 2021, 9(8): 908-908. DOI: 10.3390/jmse9080908
    [24]
    张宏群, 班勇苗, 郭玲玲, 等. 基于YOLOv5的遥感图像舰船的检测方法[J]. 电子测量技术, 2021, 44(8): 87-92. https://www.cnki.com.cn/Article/CJFDTOTAL-DZCL202108017.htm

    ZHANG Hongqun, BAN Yongmiao, GUO Lingling, et al. Detection method of remote sensing image ship based on YOLOv5[J]. Electronic Measurement Technology, 2021, 44(8): 87-92. https://www.cnki.com.cn/Article/CJFDTOTAL-DZCL202108017.htm
  • Related Articles

    [1]WANG Wenjin, KONG Jincheng, QI Wenbin, ZHANG Yang, SONG Linwei, WU Jun, ZHAO Wen, YU Jianyun, QIN Gang. Research Progress on Materials and Devices of HgCdTe p-on-n Double Layer Heterojunction Grown by VLPE[J]. Infrared Technology , 2024, 46(3): 233-245.
    [2]YANG Chunzhang, QIN Gang, LI Yanhui, LI Da, KONG Jincheng. Research on Growth of M/L-wavelength Dual-band IR-MCT on CZT Substrate by MBE[J]. Infrared Technology , 2018, 40(1): 1-5.
    [3]ZHOU Lianjun, HAN Fuzhong, BAI Piji, SHU Chang, SUN Hao, WANG Xiaojuan, LI Jinghui, ZOU Pengcheng, GUO Jianhua, WANG Qiongfang. Review of HOT MW Infrared Detector Using MCT Technology[J]. Infrared Technology , 2017, 39(2): 116-124.
    [4]QIN Gang, LI Dongsheng, LI Xiongjun, LI Yanhui, WANG Xiangqian, YANG Yan, TIE Xiaoying, ZUO Dafan, BO Junxiang. Research on the Technique of in-situ p-on-n MWIR-MCT by MBE[J]. Infrared Technology , 2016, 38(10): 820-824.
    [5]WANG Yi-feng, LI Pei-zhi, LIU Li-ming, WANG Dan-lin. Developments of Very Long Wavelength Mercury Cadmium Telluride Infrared Detectors[J]. Infrared Technology , 2012, 34(7): 373-382. DOI: 10.3969/j.issn.1001-8891.2012.07.001
    [6]Developments of Mercury Cadmium Telluride in Recent Years[J]. Infrared Technology , 2009, 31(8): 435-442. DOI: 10.3969/j.issn.1001-8891.2009.08.001
    [7]The Determination of Cadmium-Mercury Telluride Composition for Any Thickness by Infrared Transmission[J]. Infrared Technology , 2005, 27(1): 39-41. DOI: 10.3969/j.issn.1001-8891.2005.01.009
    [8]Measurement on Minority Carrier Lifetime of Mercury Cadmium Telluride Material by Microwave Photoconductivity Decay Method[J]. Infrared Technology , 2003, 25(6): 42-44,48. DOI: 10.3969/j.issn.1001-8891.2003.06.012
    [9]p+n Infrared Detectors by As Ion Implantation in HgCdTe[J]. Infrared Technology , 2002, 24(4): 46-48,26. DOI: 10.3969/j.issn.1001-8891.2002.04.012
    [10]The Surface Passivation of MCT Infrared Detectors[J]. Infrared Technology , 2001, 23(3): 9-12,15. DOI: 10.3969/j.issn.1001-8891.2001.03.003
  • Cited by

    Periodical cited type(4)

    1. 王振,刘磊. 基于改进分水岭算法的电力设备红外图像分割. 红外技术. 2025(04): 484-492 . 本站查看
    2. 刘沛津,张香瑞,魏平. 基于融合重构的电气设备红外图像EnFCM聚类分割方法. 红外技术. 2024(03): 295-304 . 本站查看
    3. 张利军. 基于红外热成像技术的变电站巡检机器人的应用. 山东煤炭科技. 2024(12): 168-172 .
    4. 冯杰,张莹,叶影,贺润平,王哲斐. NSST域电气设备红外图像增强处理算法设计. 电子设计工程. 2023(21): 176-179+185 .

    Other cited types(3)

Catalog

    Article views (583) PDF downloads (138) Cited by(7)
    Related

    /

    DownLoad:  Full-Size Img  PowerPoint
    Return
    Return