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

A Lightweight Dual Head Adaptive Infrared Weak Detection Algorithm Based on YOLO

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

     

    Abstract: Infrared detection of small targets is a prerequisite for the realisation of automated infrared intelligent security, surveillance, and guidance systems. However, infrared detection of small targets suffers from high false positive and false negative rates, as well as severe information loss. To achieve real-time, high-precision infrared detection of small targets, we propose a lightweight infrared detection algorithm based on YOLO. First, targeting the feature information of infrared small targets, a dual-detection-head lightweight network structure, E-YOLO, is proposed to address the issue of target information disappearing during feature extraction; then, a feature-space-based filter pruning algorithm is proposed, which calculates the similarity between filters to prune redundant filters and improve inference speed; Finally, the cosPIoU-Varifocal loss function is proposed, which uses an adaptive penalty factor based on target size and combines 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. The algorithm's computational load and network model size were compressed to 0.9 GFLOPs and 190 KB, respectively, with inference time below 3 ms. Compared to existing algorithms, the proposed algorithm achieves improvements in average detection accuracy, parameter count, and model size, meeting real-time detection requirements.

     

/

返回文章
返回