针对红外图像分辨率差、对比度低、信噪比低、视觉效果模糊等检测难点,提出一种基于YOLOv5的轻量化红外目标检测(Lightweight Infrared Target Detector-YOLO,LITD-YOLO)算法.LITD-YOLO算法重新设计网络结构,针对红外目标成像特征,将特征提取网络与特征融合网络结构重构.提高小感受野权重,重建浅层特征和深层特征之间的多尺度融合关系,提高浅层网络语义信息表征能力权重,增强对红外小目标的检测能力.引入Varifocal loss以实现交并比感知分类评分(Intersection over Union-Aware Classification Score,IACS)回归,使模型对密集目标的检测能力进一步加强.使用SIoU作为边框损失函数,用于提升预测框的准确度,同时加速模型收敛.实验结果表明,在FLIR和OSU数据集下模型检测精度分别提高至88.5％、99.7％,模型体积仅3.9 MB,参数量和算法复杂度大幅降低;与主流算法相比,LITD-YOLO在各项指标上均取得了不错的进步,在检测精度、模型体积和推理速度等方面具有先进性,能满足对红外目标的高质量检测.
Research on Lightweight Infrared Target Detection Algorithm
For poor resolution,low contrast,low signal to noise ratio,visual effect blurring and other detection difficulties of infrared images,a Lightweight Infrared Target Detection-YOLO(LITD-YOLO)algorithm based on YOLOv5 is proposed.Firstly,the LITD-YOLO algorithm redesigns the network structure and reconstructs the feature extraction network and feature fusion network structure for the infrared target imaging features.The small sensory field weight is increased to reconstruct the multi-scale fusion relationship between shallow and deep features,and the semantic information representation capability weight of the shallow network is increased to enhance the detection capability of small infrared targets;then,Varifocal loss is introduced to achieve Intersection over Union-Aware Classification Score(IACS)regression to further enhance the detection capability of the model for dense targets;finally,SIoU is used as the border loss function to improve the accuracy of the prediction frame while accelerating the convergence of the model.The experimental results show that the model detection accuracy improves to 88.5％and 99.7％for FLIR and OSU datasets respectively,and the model volume is only 3.9 MB,with a significant reduction in the number of parameters and algorithm complexity;compared with mainstream algorithms,LITD-YOLO has made good progress in all indicators,and is advanced in terms of detection accuracy,model volume and inference speed,and can meet the requirements for high quality detection of infrared targets.
万方数据
维普
