低空轻量级红外弱小目标检测算法.pdf

第 卷 第 期 激 光 与 红 外 ，年 月 ，文章编号：（）图像与信号处理低空轻量级红外弱小目标检测算法张上，黄俊锋，王恒涛，陈永麟，王康，（三峡大学湖北省建筑质量检测装备工程技术研究中心，湖北 宜昌 ；三峡大学计算机与信息学院，湖北 宜昌 ）摘要：精准的红外弱小目标检测是实时监控、追踪、制导的关键；红外弱小目标存在检测难度高、误检高、漏检严重的问题。为了提高红外弱小目标检测算法的实时性和检测精度，提出了一种超轻量红外弱小目标检测算法 。首先，重设计下采样方案，针对红外图像特征信息调节网络架构，解决红外弱小目标特征梯度降低和特征消失问题；然后设计网络模型剪枝算法，实现剪枝算法与网络结构的融合，去除冗余参数，实现检测速度的提高；最后设计 损失函数，在均衡正负样本与重叠损失的同时，对正样本进行加权处理，解决背景干扰问题。实验结果表明，在 和 数据集下检测精度分别提高至 、，模型体积和计算量可压缩至 、，推理速度降至 以下。与主流算法进行对比，改进后算法在检测精度、模型体积、计算量等方面均取得了不错的成绩。能够满足实时性检测需求。关键词：目标检测；模型剪枝；中图分类号：；文献标识码：基金项目：国家级大学生创新创业训练计划项目（；）资助。作者简介：张上，男，博士，副教授，主要研究方向为物联网技术，计算机应用技术。：通讯作者：王恒涛，男，硕士研究生，主要研究方向为计算机技术，目标检测技术。：收稿日期：，（，；，）：，；：；引言红外目标检测是实时监控、追踪、制导的关键技术之一，其应用广泛，如机载监视、反无人机等领域。目前，红外弱小目标检测成为计算机视觉中的一个重点与难点，红外图像弱小目标检测的难点主要有：红外弱小目标亮度低，通常呈点状，成像不明显；尺寸微小，一般不超过 个像素，形状特征、纹理和边缘特征难以提取；背景复杂，在实际检测环境下，背景因素不可避免，如楼房、云层、植物、飞行动物等，都会干扰检测；红外图像成像存在噪声；实时性要求高。如监测与制导场景中，需要检测算法能够及时响应目标，提供高准确率和高速的检测。上述难点亟待深度学习来解决。基于深度学习的目标检测算法可分为四类，分别为注意力机制算法（和 等）、双阶段检测算法（、等）、单阶段检测算法（、等）和无锚框检测算法（、等）。人工智能算法广泛用于工业和军事 。为实现红外弱小目标的智能化 实 时 检 测，王 晓 红 等 人 提 出 一 种 基 于 的红外弱小目标检测方法，引入 特征融合机制，使网络模型可以更聚焦于物体的细小特征。张立国等人 对特征融合模块和解耦检测端分类回归结构进行改进。以上算法能够实现红外弱小目标的检测，但存在检测误检率、漏检率高，计算量和模型体积大的问题。针对红外弱小目标检测存在的难点，本文贡献如下：）设计专用于红外弱小目标检测的 （超轻量红外目标检测）算法，对网络结构进行重构，使其能够适应红外弱小目标特征提取，并且减轻算法的计算量与体积。）设计重构网络的剪枝架构，实现剪枝算法与重构后网络的融合。去除冗余参数与通道，提高模型的推理速度。）设计 损失函数。改变边框回归损失函数、均衡正负样本、提高检测框检测精度。）为验证算法有效性，在 和 公开数据集上对算法进行验证。在数据集上的平均检测精度分别为 、。结构设计 核心思想是针对红外弱小目标特征进行网络结构的构建，以红外弱小特征信息的提取与融合为基准点，分别设计网络特征提取模块与网络特征融合模块，然后设计模型剪枝算法，实现剪枝算法与网络模型的融合，最后，设计 损失函数，实现正负样本均衡与重叠区域的损失计算，提升与修复剪枝完成后的精度。主要由轻量化网络结构、剪枝算法和 损失函数构成。网络架构如图 所示，通过在轻量化提取网络与融合网络的训练过程中融合剪枝算法和 损失函数，完成模块轻量化和回归优化。图 系统架构图 经过轻量化设计后，实现网络结构冗余模块的剔除，完成冗余参数的修剪，使得网络模型计算量、体积大幅减低，检测速度提高。网络重构目标检测中常使用 、和 三类特征图，通过调整感受野和特征图来实现红外弱小目标的特征提取和特征融合。针对红外弱小目标的特征信息与尺度变化，将感受野调整激 光 与 红 外 张上等低空轻量级红外弱小目标检测算法为 与 两种特征尺度，将更多红外弱小目标信息保留。在红外图像中，红外弱小目标仅占据图片的很小部分，经过卷积与下采用后信息丢失严重。优化后特征提取网络 结构如表 所示，优化后特征提取网络下采样为 次，卷积为 次，在降低计算量和参数量的同时保留更多的红外弱小特征信息。表 重构 结构 （，）（，）（，）（，）次下采样 特征提取网络经过优化后需要对特征融合网络作处理，改进后特征融合网络结构如图 所示，针对感受野尺度调整融合网络结构。依次完成浅层到深层语义的融合，再进行深层到浅层语义的融合。同时，增加浅层语义的特征权重。减少红外弱小目标的信息丢失。图 特征融合网络结构图 算法结构如图 所示，网络结构重构后，分别对特征提取网络与特征融合网络进行了重构设计。重构后网络结构删除大感受野信息提取，仅保留中小尺度的特征提取，特征融合网络侧重向小感受野进行调整，最后输出 与 特征信息。图 网络结构 剪枝算法融合基于范数的剪枝算法是被经常使用的模型剪枝算法，通过特征信息的范数大小来均衡重要程度。在目标检测特征空间中低范数特征信息也具有重要的特征信息，而基于范数的剪枝算法将其删除，故丢失大量低范数特征信息。剪枝算法如图 所示，将剪枝算法融入到模型中，突破常规范数的局限性，去除可替代的特征信息，保留更多有用信息。图 剪枝示意图 的核心是计算滤波器的欧氏距离之和，过滤出欧式距离之和相对较低的进行删除。计算滤波器 （）的欧式距离：()，（）过滤出可代替的滤波器：（）（）使用几何中值的算法计算第 层中所有滤波器的公共 距 离。其 中，滤 波 器 参 数 定 义 为，。，（）激 光 与 红 外 第 卷滤波器根据几何中心距离进行剪枝，找出距离特征空间中心最近的滤波器：，（）几何中心距离的计算涉及到范数的计算，导致计算量的增加。而计算滤波器之间的距离相对简单，故可以通过计算距离之和得到几何中心相对较小滤波器之和：，（）（），（）其中，。算法 前期工作：训练集、验证集、预训练模型输入，参数设定剪枝率设定：网络模型及参数 ：；：迭代网络模型：；：计算第 层中每个滤波器之间的欧氏距离之和：得到 个可替换滤波器：将可替换滤波器梯度置零：训练模型 调节梯度获得待去零模型 ：去除 网络中梯度为零的通道，获得去零网络 输出：剪枝且模型收敛的网络与参数 通过对 进行改进，设计 在提高锚框匹配的同时根据目标类别进行加权损失计算。、损失函数如下：（，）（）（）（）（）（）（，）（）（）（）（）（）其中，为预测值，取值为 ，；为实际背景与目标类别；为比例因子；（）为目标类调制因子；为背景类调制因子；为判别条件，背景类 为，目标类 值为真实锚框与预选框之间的 值。通过引入目标调节因子，使用加权的方式可以使得 回归，解决目标类与背景类之间的不均衡问题。但对于目标类采用的是相同的计算方式，不能对正样本进行突出。基于二元交叉熵损失进行设计，使用真实锚框与预选框之间的 值进行损失计算，使得在均衡正负样本的同时能够根据样本的质量进行损失加成。常见的边框损失忽视所需真实框与预测框之间不匹配的方向问题。故提出 ，既提高了训练速度，又提高了检测精度。损失函数如下：（）其中，为 损失函数；为距离损失；为形状损失。形状损失 通过锚框的宽高进行处理：，()（）其中，为形状损失调节因子；通过调节因子计算形状损失，能够直接对宽高损失进行调节。距离损失 可依靠角度、横纵坐标得出：，（）()，()（）以锚框横纵坐标、角度为基准，通过计算出距离误差，其中，为角度误差：()()（），在均衡正负样本、重叠损失、锚框向量角度的同时，重点突出正样本的贡献：（）（）系统算法详解 通过将剪枝算法、重构网络模型训练、损失函数优化相结合，完成轻量化红外弱小目标检测算法的设计与实现。模型训练及剪枝优化如算法 所示。实验实验 环 境：操 作 系 统：、；配置：，显存；深度学习框架：。红外弱小目标数据集红外小目标在学术上有两种定义方式，分别为基于相对尺度和基于绝对尺度的定义方法。基于相对尺度的定义方法，是根据国际光电仪器工程师协会的定义，将小于原图 像素的目标称为小目标。基于绝对尺度的定义方法，根据通用目标检测激 光 与 红 外 张上等低空轻量级红外弱小目标检测算法数据集 数据集中，尺寸小于 像素的目标定义为小目标。算法 算法准备工作：训练参数、预训练模型输入，训练次数 剪枝率：（剪枝算法未收敛 ）网络利用 、结构提取特征信息，使用 激活函数对 结构进行优化。()()（()）网络使用上采样与连接进行尺度变化，上采样完成尺度的匹配，完成特征融合。计算融合特征：（）预测层利用 突出正样本权重。进行锚框定位。对生成的锚框进行阈值处理。损失函数：()()（）网络模型剪枝收敛，去除梯度为 的卷积核、参数、通道。输出：收敛的轻量化网络模型为验证 对红外弱小目标的检测效果，本文采用公开数据集 和 数据集进行验证。是一种单帧红外小目标检测数据集，共包含 张图片，包含 个实例，数据量丰富，包含短波、中波和 波长的红外图像。数据集有 种背景，覆盖地面复杂背景、天空、地空交接等复杂背景。训练集、验证集、测试集按照 进行随机划分。模型剪枝对比模型剪枝性能对比如表 所示，其中剪枝模型为网络重构后的模型，剪枝率的增加，模型的检测性能有所下降。网络结构重构后在 和 数据集上进行验证，剪枝率升高，模型的检测精度 有所减低，当剪枝率大于 时，模型的检测精度下降最多。针对剪枝带来的精度下降，本文在模型剪枝后设计 进行算法微调。微调结果如表 所示，从表中数据可知，微调后模型的 、和 均有恢复。其中，在剪枝率为 时，性能恢复最多。其中，在 数据集检测精度能提高 ，在 数据集检测精度能提高 。实验充分验证了 算法中的 有效性。表 不同剪枝率对数据集的性能指标对比 实验结果与分析为验证 中每个改进点对红外弱小目标检测的效果，在 、数据集上进行消融实验，如表 所示。经过网络结构重构、优化后平均检测精度、准确率、召回率均有所提高。在两个不同的数据集上，通过网络重构与 后，能够大幅提高召回率和准确率。随着模型剪枝的提升，模型平均精度精度逐渐下降，经过 可恢复检测精度。激 光 与 红 外 第 卷表 不同剪枝率经过 微调后性能指标比较 表 消融实验 算法重构 剪枝 基准 与先进算法对比为 验 证 算 法 对 红 外 弱 小 目 标 检测的先 进 性，对 进 行 对 比 实 验，将 与当前主流算法进行对比，如表 和表所示。算法在当前主流算法具有竞争力。算法还可以根据剪枝率的大小进行计算量、推理时间和体积缩减，能够满足不同应用环境进行调节，满足不同环境下的实时检测需求。总的来说，算法在模型体积、计算量、平均检测精度和推理时间上均有巨大优势，能够满足红外弱小实时目标检测需求。表 不同算法对 的性能指标比较 参数量 （）激 光 与 红 外 张上等低空轻量级红外弱小目标检测算法表 不同算法对 的性能指标比较 参数量 （）实验效果与分析为验证 算法对红外弱小目标的实际检测效果，在 、数据集进行验证，包含海上、陆地、飞机等复杂背景。图 分别为不同剪枝率在海上、陆地、飞机的检测效果。如图 所示，（）为 基线检测结果图、（）为 检测结果图、（）为 检测结果图。相比于当前主流算法检测置信度更高，漏检率和误检率更低。图 不同模型下的检测效果 图 图 从左到右依次为剪枝率、。在海上、陆地、飞机检测效果均优秀；随着剪枝率的增加，红外目标检测不会存在漏检率提高的问题，但置信度存在降低。在剪枝率 以下，置信度保持较高水平；当剪枝率超过 时，实例检测置信度下降较多。图 不同剪枝率在海上红外目标的检测效果 综上所述，对海上、陆地、飞机均有优异的检测效果。在剪枝率较高的情况下，模型的计算量与体积均大幅下降，漏检率和误检率不会增加，但存在置信度降低的问题。图 不同剪枝率在陆地背景红外目标的检测效果 图 不同剪枝率对飞机红外检测效果 结论精准的红外弱小目标检测对红外遥感至关重要。红外弱小目标特征信息尺寸小、背景复杂导致其检测难度增加。针对红外弱小目标检测特征，本文设计 算法。为保存更多特征信息，降低卷积带来的特征信息丢失问题，重设计网络结构，在兼顾检测效果的同时，模型更加轻量。然后设计模型剪枝算法，实现冗余特征信息的剔除，提高检测速度，降低计算量、模型体积和参数量；最后设计 损失函数，改变边框回归损失函数、均衡正负样本、提高检测框检测精度。在 、数据集上进行算法验证，各项指标均优于当前主流算法，取得了优秀的成绩。算法精度最高分别可达 、，模型体积最小为 ，推理时间最低为 。在平均检测精度、推理速度与模型体积均具有较大的先进性。激 光 与 红 外 第 卷参考文献：，：，：，：，：，：，（）：，：，：，：，：，：，：，：，：，（）：（）王恒涛，张上，张朝阳等 基于 的轻量化 缺陷检测 无线电工程，（）：，（）：（）张上，王恒涛，冉秀康 基于 的轻量化交通标志检测方法 电子测量技术，（）：，（）：，（）王恒涛，张上 轻量化 图像舰船目标检测算法 电光与控制，（）：，（）：（）王晓红，陈哲奇 基于 算法的红外图像行人检测研究 激光与红外，（）：，（）：（）张立国，马子荐，金梅等 基于 的轻量红外图像行人 检 测 方 法 激 光 与 红 外，（）：激 光 与 红 外 张上等低空轻量级红外弱小目标检测算法