基于改进YOLOv3的机场盲区遥感图像目标检测方法_杨轲.pdf

第 卷第期 年月 科技和产业 ，基于改进 的机场盲区遥感图像目标检测方法杨轲，董兵，吴悦，郝宽公，耿文博（中国民用航空飞行学院 空中交通管理学院，四川 广汉 ）摘要：针对机场存在的雷达监视盲区问题，提出一种改进 目标检测算法。首先，基于原 主干特征提取网络加入 池化模块，以深度可分离卷积替代普通卷积。然后，针对小尺度目标数据集，增加第层金字塔加强特征提取网络，在 聚类算法的基础上提出一种线性放缩进行锚框筛选。最后，在 数据集上 为 ，为 。仿真结果表明该方法可以满足机场实时目标检测任务的需求。关键词：遥感图像；多特征融合；线性放缩；深度可分离卷积；实时目标检测中图分类号：文献标志码：文章编号：（）收稿日期：作者简介：杨轲（），男，山西长治人，中国民用航空飞行学院空中交通管理学院，硕士研究生，研究方向为航空运行管理；董兵（），男，四川德阳人，中国民用航空飞行学院空中交通管理学院，教授，博士，研究方向为航空运行管理；吴悦（），女，山西临汾人，中国民用航空飞行学院空中交通管理学院，硕士研 究 生，研究方向为航空运行管理；郝 宽公（），男，山东临沂人，中国民用航空飞行学院空中交通管理学院，硕士研 究 生，研究方向为航空运行管理；耿 文博（），男，吉林松原人，中国民用航空飞行学院空中交通管理学院，硕士研究生，研究方向为航空运行管理。遥感目标检测旨在遥感图像中获取重要目标的位置、种类等语义信息，在交通流量监测、交通疏导等领域中扮演了重要的角色。航空遥感数据包含了丰富的语义信息，通过识别遥感图像获取客机位置、态势等信息，将在很大程度上提升民航运行的安全性。在实际运行中，过远台的进近航空器会在雷达屏幕上消失 ，这一段视野盲区将对民航安全运行带来隐患。过去由于卫星技术的限制，针对遥感图像的目标检测技术难以发展。随着北斗全球卫星导航系统正式运营，对地高分辨率观测系统相继建成，相关民用空间基础设施不断完善，利用遥感数据实时进行数据分析成为可能。为解决上述雷达进近盲区的问题，甚至实现智慧高效监测的目标，亟须将人工智能技术应用到民航监视系统的实际运行中。随着 和 的提出，推动了深层神经网络和计算机视觉的发展。目前发展较为成熟的深度学习目标检测方法主要有两类：一类是 、等两阶段方法，检测需要先提取检测区域再分类回归，具有速 度 较 慢 但 精 度 较 高 的 特 点；另 一 类 是 、等 一 阶 段 方法，通过直接回归的方式进行目标检测，精度相对低但速 度 快 很 多。李 靓 等在 遥 感 图 像 中 采 用 处理小目标检测。文献 针对遥感影像构建多特征融合 网络，实现不同特征遥感图像的目标检测。文献 提出支持向量机融合的检测方法。不难看出，随着残差网络、密集网络的提出与计算机硬件设施不断地完善，遥感图像利用深度卷积神经网络识别与检测越来越准确。文献 针对电力场景小目标漏检错检问题，在采用多尺度变换方法的基础上结合拉普拉斯算法进行锐化处理，提 高 了 模 型 在 监 控 视 频 下 的 检 测 精 度。文献 在传统 算法上提出使用动态目标的坐标偏移量进行先验框回归运算的方法，对常规目标检测算法多目标锚框难区分的情形作出了改进，提高了机场场面目标的检测精度和检测效率。文献 提出将轻量级网络与金字塔层级网络进行特征融合。文献 将 中的特征提取网络替换为 以减少网络参数量。解决终端区盲区航空器检测的问题对检测速度提出了更高要求，因此选用一阶段法 对 航 空 器 目 标 进 行 检 测。主 要 工 作 有：利 用 算 法 结合线性放缩的方式筛选先验框，针对小目标数据集加入 特征提取结构，以深度可分离卷积替代普通卷积并构建层金字塔加强特征提取网络等方式提高模型对小尺度目标的识别能力。算法简介 网络结构 的网络结构包括主干特征提取网络和加强特征提取网络两个部分，如图所示。图 网络结构主干特征提取网络以 为主，有层卷积层和层残差层。原始输入 ，经过层输出通道为、卷积核大小为、为、为的卷积，将得到的 图片放入残差层。每层残差层包含卷积下采样与堆叠残差块两个步骤。残差块是依次进行、两次卷积、正则与 激活的结构，堆叠是指残差块数量按照、进行逐层堆叠。加强特征提取网络以主特征提取网络底层的输出作为输入，分别是 、，对位于最底层的 输入，通过次、调整通道数的交替卷积，可减少参数并得到 的图像，一个方向传播是进行两次卷积便得到最底层的预测输出，其 为 ，另一个是进行两倍上采样得到 为 的图像，目的是为了在保留对大目标检测 特 征的 基 础 上 与 输 入 的 图像进行多尺度特征融合，其余各层同理按步骤进行特征提取和融合，进而得到层金字塔多尺度特征融合的结果，分别为 、（感受野由大到小），代表（），即坐标、置信度与种类数。数据处理流程首先在数据传入网络前进行 转换以便输入图像的数据处理，再进行维度转置并添加 维度实现归一化以满足网络尺度要求。其次筛选出某一区域内同类值最大的锚框并进行预测结果的堆叠。最后为得到最终的边界框，需要对预测框的位置进行调整，得出 、这个尺度分别对应在图像的位置，通过非极大抑制，即从种类置信度和物体置信度两方面判断是否大于阈值需求。算法改进 先验框筛选改进关于先验框的选择，原模型通过 聚类算法进行筛选，可以聚类出个效果不错的锚框，但 聚类存在下述局限性：初始聚类中心的个数难以确定，必须通过人工给定，导致聚类结果比较分散，且容易陷入局部最优。为减少确定初始聚类中心的主观性、增加筛选先验框的随机性，提出改进 对数据集进行初筛，算法：随机选取数据集中的一个样本为质心，即初始聚类中心；计算其余样本到已知聚类中心的欧氏距离，通过轮盘赌算法计算欧氏距离平方占比并推导下一个聚类中心的概率；通过对上述步骤进行迭代，直到获取个聚类中心；将得到的个聚类中心执行标准 算法。模型改进 目标检测采用的 数据集尺度较大，而航空遥感图像大多是小目标。为解决这一问题，首先本文在原层加强特征提取网络的基础上增加第层特征提取尺度，通过对原 的输出继续进行倍上采样，与主特征提取网络的输出进行特征融合，构建层金字塔特征网络结构，提高网络对小目标的检测能力。其次，基于 网络加入 特征提取结构，通过混合池化加强对图像特征的提取能力。最后在 结构后引入深度可分离卷积替代普通卷积，目的是在提升网络检测精度的基础上尽可能减少网络参数、提升网络运算速度，如图所示。激活函数会压缩卷积权值海量的特征信息，导致模型的表达水平急剧下降。通过限制权值范围，可以更均匀地映射权值信息，使得权值特征更大可能被保留，因此 有效改善了 权值范围相差过大的问题，很大程度上减少了量化误差。科技和产业 第 卷第期图改进 网络结构 模型训练采用初始学习率为 、批次数为、数为 、衰减系数为 ，以及动量为 的异步梯度随机下降进行训练。深度神经网络模型的训练通常都会使用原始的预训练权重，是因为原模型采用的数据集样本数远大于个人数据集，从零开始训练可能造成权重过于随机而导致特征提取效果变差。因此对算法的网络结构进行了重写，不直接迁移学习网络的预训练权值，通过主干网络的预训练权重进行训练，通过充足的训练来避免陷入局部最优。考虑到计算机 运算性能及显存容量的限制，将训练拆分为冻结与解冻两个过程。冻结的 轮数设置为，冻结是指网络训练时不更新指定层中的权值，目的在于减少模型网络的训练和收敛的所需时间、避免梯度爆炸的可能，解冻的操作就是激活所有被冻结的进程，重启原先执行的任务，通常在 时使用。实验及结果 运行环境与数据集算法的运行环境为 ，显卡，系统版本 ，数据集为武 汉 大 学 标 注 的 公 开 遥 感 数 据 集 ，航空遥感数据集包含 张图共 架飞机，其中图像复杂背景、高曝光、低能见度和小目标等情况如图所示。从数据集角度看，直接使用目标检测模型的识别精度较低，是因为遥感数据集具有以下个层面的特殊性：背景复杂性，遥感图像通常具有相图航空数据集实例对大的视野，易混入部分干扰物体导致漏检误检的发生；视角特殊性，传统网络模型的数据集的采样视角基本都是平视拍摄，遥感图像的采样视角通常是 在高 空 进行 俯视；小 目 标 问 题，如 图 所示，利用对航空遥感数据集进行聚类发现图像的目标通常都是小目标，通常只有几到几十像素且集中，与常规数据集分散的多尺度图像相比特征提取不显著。评估参数判断目标检测的好坏主要取决于精确率、召回率、值和值。精确率，又名查准率，即在全部预测正确集合中正确预测为正所占比例，反映检测准度；召回率，也称查全率，是在全部实际正确集合中正确预测为正的比例；值是 曲线下的面积，用以衡量模型的表现；平均正确率是一个直观体现检测性能的数值，是精确度和召回率的调和平均值。杨轲等：基于改进 的机场盲区遥感图像目标检测方法图 聚类结果 （）（）（）（）（）式中：表示正类被选中；代表负类被选中；表示 正 类 未 选 中；为 精 确 率；为 召 回 率。用 代表平均精度。消融实验设计表所示的消融实验，目的在于分析不同种类 对目标检测精度的影响。表 算法对比模型原模型层网络 线性放缩 表中，模型代表原 的复现算法，向下依 次表示在 算法 基 础 上采用 不同 后的模型，按列依次为层加强特征提取网络、线性放缩、结构。表显示，采用层网络结构后模型的识别能力反而出现下降，这是因为航空遥感数据集目标尺度相对较小，而常规采用聚类得到的先验框类别单一，尺寸比例的分布相对集中，导致直接聚类生成的预测锚框存在精确率低、多漏检等训练效果较差的情况。针对小目标数据集不同的特征层应分配相应不同尺寸的先验框，例如 越小的特征层适合越大的先验框，提出面向小目标数据集的线性放缩算法。表消融实验结果模型标签类别 如图所示，提出一种对真值锚框的宽高同步进行拉伸，保持真值框中心点的坐标不改变的线性缩放的缩放方式，使锚框离散在更多不同且有间隔的尺度上。图线性放缩结果将改进后的算法称为 ，公式为?（）?（）?（）（）（?）?（）?（）式中：?、?表示放缩后坐标；、为变换前的坐标；据经验取 、线性放缩。如表，相较于原 算法，采用聚类和线性放缩的策略后的模型的精确率提高了 ，召回率提高了 ，提高了 ，提高了 。实验表明这种线性放缩的改进使模型泛化性得到提升，为改善小尺度数据集的目标检测效果提出一种新的解决思路。最后，采用 和深度可分离卷积后算法的目标检测各项指标均有提升，为 。算法对比为进一步验证本文提出算法的有效性，在 航空遥感数据集上对比 、和本文提出的 算法的性能表现（表）。表算法评估结果算法 科技和产业 第 卷第期表显示，第列表示对比采用的不同算法，依次为 、和本文算法，它们的 依 次 为 、和 ，其中 的检测精度较高，但归因于其两阶段算法本身的限制，导致 仅有 ，未能达到实时检测的标准（）。算法的识别效果较差，只有 ，原因在于该算法为应对不同尺度的检测任务，设计了个不同大小的特征层进行目标识别，但原 模型缺乏多尺度特征融合的网络结构，造成深层网络语义信息丰富而浅层匮乏的问题，进而使得算法对小目标检测能力差。目标检测算法在检测精度和检测速度上都有不错的表现，但针对小尺度目标检测能力不足，在此基础上，本文提出的 算法通过线性放缩等多种 改善了模型对小目标的识别精度，使 达到了 ，较原算法提高了 ，虽然网络加深等原因造成 值下降了 ，仍达到实时检测标准，可以满足机场盲区航空器目标检测需求。在上述实验条件下，绘制了原 与本文算法 的平均损失函数曲线（图）。本文提出的检测模型的训练损失在前 次呈现指数下降趋势，轮下降趋势趋于平缓，轮后损失函数的值稳定在 左右，算法在 左右，模型收敛速度显著超过原算法。该实验结果表明，本文所提的改进 目标检测算法具有较好的效果。图损失函数曲线如图所示，为了更加直观地体现检测效果，将原始 和本文算法 的识别结果进行对比。本文算法在低能见度、复杂背景、高曝光及小目标等场景下的识别效果均优于原算法，显著改善原算法出现的误检、漏检状况，且在置信度上也有相当程度的提升，可知本文算法改进的有效性，且提高了对复杂场景下目标的检测能力。图遥感图像检测对比 数据增广评估基