低空轻量级红外弱小目标检测算法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家级大学生创新创业训练计划项目(No.202111075019;No.202011075013)资助。


Low altitude lightweight infrared weak small target detection algorithm
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    精准的红外弱小目标检测是实时监控、追踪、制导的关键;红外弱小目标存在检测难度高、误检高、漏检严重的问题。为了提高红外弱小目标检测算法的实时性和检测精度,提出了一种超轻量红外弱小目标检测算法SL YOLO。首先,重设计下采样方案,针对红外图像特征信息调节网络架构,解决红外弱小目标特征梯度降低和特征消失问题;然后设计网络模型剪枝算法,实现剪枝算法与网络结构的融合,去除冗余参数,实现检测速度的提高;最后设计Varifocal SIoU损失函数,在均衡正负样本与重叠损失的同时,对正样本进行加权处理,解决背景干扰问题。实验结果表明,在SIRST和IDSAT数据集下检测精度分别提高至964、981,模型体积和计算量可压缩至190 kB、09 GFLOPs,推理速度降至3ms以下。与主流算法进行对比,改进后算法在检测精度、模型体积、计算量等方面均取得了不错的成绩。能够满足实时性检测需求。

    Abstract:

    Accurate infrared small and weak target detection is the key to real time monitoring,tracking,and guidance.Infrared weak and small targets have problems of high detection difficulty,high false detection,and serious missed detection.In this paper,an ultra lightweight infrared dim small target detection algorithm SL YOLO is proposed to improve the real time performance and detection accuracy of infrared dim small target detection algorithms.Firstly,the downsampling scheme is redesigned to adjust the network architecture for the infrared image feature information to solve the problem of feature gradient reduction and feature disappearance for infrared weak targets.Then,a network model pruning algorithm is designed to integrate pruning algorithm with network structure,removing redundant parameters,and improving detection speeds.Finally,the SIoU Varifocal loss function is designed to equalise the positive and negative samples with overlapping losses while weighting the positive samples to solve the problem of background interference.The experimental results show that the detection accuracy is improved to 96.4% and 98.1% under the SIRST and IDSAT datasets,respectively.The model volume and computational complexity can be compressed to 190 kB and 0.9 GFLOPs,and the inference speed is reduced to less than 3 ms.Comparing with the mainstream algorithms,the improved algorithm has achieved good results in terms of detection accuracy,model volume;computational complexity.It can meet the real time detection requirements.

    参考文献
    相似文献
    引证文献
引用本文

张上,黄俊锋 ,王恒涛,陈永麟,王康.低空轻量级红外弱小目标检测算法[J].激光与红外,2024,54(1):122~129
ZHANG Shang, HUANG Jun-feng, WANG Heng-tao, CHEN Yong-lin, WANG Kang. Low altitude lightweight infrared weak small target detection algorithm[J]. LASER & INFRARED,2024,54(1):122~129

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-01-23
  • 出版日期: