基于YOLOv5的轻量双头自适应红外弱小目标检测算法

Lightweight Dual Head Adaptive Infrared Weak Target Detection Algorithm Based on YOLOv5

  • 摘要: 红外弱小目标检测是实现自动化红外智能安防、监控、制导的前提条件;红外弱小目标检测存在误检漏检率高、信息丢失严重等问题。为了实现实时高精度红外弱小目标检测,提出了一种基于YOLOv5的轻量级红外弱小目标检测算法。首先,针对红外弱小目标特征信息,提出了双检测头轻量级网络结构E-YOLO,解决目标信息在特征提取过程中消失问题;然后,提出了基于特征空间的滤波器剪枝算法,通过计算滤波器之间的相似度,裁剪冗余滤波器,提高推理速度;最后提出了cosPIoU-Varifocal损失函数,使用目标尺寸自适应惩罚因子,结合角度平衡,在平衡样本和重叠区域损失的同时,完成正样本加权处理,解决冗余锚框干扰问题。最终,在SIRST和IDSAT数据集下进行验证,mAP为96.6%、98.2%,算法计算量和模型体积可分别压缩至0.5 GFLOPs、0.19 MB,推理时间为3 ms以下。所提出的算法在平均检测精度、参数量和模型体积等方面均获得了改善,能够满足实时性检测需求。

     

    Abstract: The infrared detection of small targets is a prerequisite for realizing automated infrared intelligent security, surveillance, and guidance systems. However, the infrared detection of small targets suffers from high false-positive and false-negative rates, as well as severe information loss. We investigated a lightweight infrared detection algorithm based on YOLOv5 to achieve real-time high-precision infrared detection of small targets. First, targeting the feature information of small infrared targets, a dual-detection-head lightweight network structure (E-YOLO) was developed to address the issue of target information disappearing during feature extraction. In addition, a feature-space-based filter pruning algorithm was used to calculate the similarity between filters to prune redundant filters and improve the inference speed. Finally, a cosPIoU-varifocal loss function was developed, which used an adaptive penalty factor based on the target size and combined angle balancing to balance sample and overlap region losses, while performing positive sample weighting, thereby addressing the issue of redundant anchor box interference. Finally, validation was conducted on the SIRST and IDSAT datasets, achieving mAP values of 96.6% and 98.2%, respectively. After compression, the computational complexity and model size of the algorithm could be reduced to 0.5 GFLOPs and 0.19 MB, respectively, with an inference time of less than 3 ms. The developed algorithm comprehensively improved the average detection accuracy, parameter quantity, and model size, and could satisfy the requirements of real-time detection.

     

/

返回文章
返回