基于改进YOLOv5的遥感小目标检测网络.pdf

第 49卷 第 9期2023年 9月Computer Engineering 计算机工程基于改进 YOLOv5的遥感小目标检测网络李嘉新1，2，3，侯进2，3，盛博莹1，2，3，周宇航2，3（1.西南交通大学 计算机与人工智能学院，成都 611756；2.西南交通大学 信息科学与技术学院智能感知智慧运维实验室，成都 611756；3.西南交通大学 综合交通大数据应用技术国家工程实验室，成都 611756）摘要：受遥感图像背景复杂、分辨率高、有效信息量少等因素影响，现有目标检测算法在检测小目标过程中存在错检、漏检等问题。提出基于 YOLOv5的遥感小目标检测算法 YOLOv5-RS。为有效减少图像中复杂背景和负样本的干扰，构建并行混合注意力模块，采用卷积替换全连接层和移除池化层的操作来优化注意力模块生成权重特征图的过程。为获取和传递更丰富且更具判别性的小目标特征，调整下采样倍数并在模型训练过程中增加小目标信息丰富的浅层特征，同时设计卷积与多头自注意力相结合的特征提取模块，通过对局部和全局信息进行联合表征以突破普通卷积提取的局限性，从而获得更大的感受野。采用 EIoU 损失函数优化预测框与检测框的回归过程，增强小目标的定位能力。在遥感小目标数据集上进行实验以验证该算法的有效性。实验结果表明，与 YOLOv5s相比，该算法在参数量减少 20%的情况下平均检测精度提升 1.5个百分点，其中，小车类目标的平均检测精度提升3.2个百分点；与 EfficientDet、YOLOx、YOLOv7相比，该算法能有效兼顾检测精度和实时性。关键词：遥感小目标检测；改进 YOLOv5；并行混合注意力；全局特征融合；损失函数开放科学（资源服务）标志码（OSID）：源代码链接：https：/ J.计算机工程，2023，49（9）：256-265.英文引用格式：LI J X，HOU J，SHENG B Y，et al.Remote sensing small object detection network based on improved YOLOv5 J.Computer Engineering，2023，49（9）：256-265.Remote Sensing Small Object Detection Network Based on Improved YOLOv5LI Jiaxin1，2，3，HOU Jin2，3，SHENG Boying1，2，3，ZHOU Yuhang2，3（1.School of Computing and Artificial Intelligence，Southwest Jiaotong University，Chengdu 611756，China；2.Laboratory of Intelligent Perception and Smart Operation and Maintenance，School of Information Science and Technology，Southwest Jiaotong University，Chengdu 611756，China；3.National Engineering Laboratory of Integrated Transportation Big Data Application Technology，Southwest Jiaotong University，Chengdu 611756，China）【Abstract】In remote sensing imagery，the detection of small objects poses significant challenges due to factors such as complex background，high resolution，and limited effective information.Based on YOLOv5，this study proposes an advanced approach，referred to as YOLOv5-RS，to enhance small object detection in remote sensing images.The presented approach employs a parallel mixed attention module to address issues arising from complex backgrounds and negative samples.This module optimizes the generation of a weighted feature map by substituting fully connected layers with convolutions and eliminating pooling layers.To capture the nuanced characteristics of small targets，the downsampling factor is tailored，and shallow features are incorporated during model training.At the same time，a unique feature extraction module combining convolution and Multi-Head Self-Attention（MHSA）is designed to overcome the limitations of ordinary convolution extraction by jointly representing local and global information，thereby extending the models receptive field.The EIoU loss function is employed to optimize the regression process for both prediction and detection frames to enhance the localization capacity of small objects.The efficacy of the proposed algorithm is verified via experiments on datasets comprising small target remote sensing images.The results show that 基金项目：国家重点研发计划（2020YFB1711902）。作者简介：李嘉新（2001），女，硕士研究生，主研方向为深度学习、目标检测；侯 进（通信作者），副教授、博士；盛博莹、周宇航，硕士研究生。收稿日期：2022-10-08 修回日期：2022-11-17 Email：图形图像处理文章编号：1000-3428（2023）09-0256-09 文献标志码：A 中图分类号：TP391第 49卷 第 9期李嘉新，侯进，盛博莹，等：基于改进 YOLOv5的遥感小目标检测网络compared with YOLOv5s，the proposed algorithm has an average detection accuracy improvement of 1.5 percentage points，coupled with a 20%reduction in parameter count.Particularly，the proposed algorithms average detection accuracy of small vehicle targets increased by 3.2 percentage points.Comparative evaluations against established methodologies such as EfficientDet，YOLOx，and YOLOv7 underscore the proposed algorithms capacity to adeptly balance the dual objectives of detection accuracy and real-time performance.【Key words】remote sensing small object detection；improved YOLOv5；parallel mixed attention；global feature fusion；loss functionDOI：10.19678/j.issn.1000-3428.00659350概述 近年来，遥感图像目标检测在无人机巡检1、农业监测2、城市规划3、生态保护4等领域得到了广泛应用，因此，对遥感图像检测的进一步优化具有重要意义。遥感图像采用空中设备对地采集信息，可以解决地面采集覆盖范围有限、减少目标被遮挡等问题。但是，从远距离和俯瞰视角拍摄的目标在高分辨率下呈现多邻域聚集、小目标占比高等特性，导致检测精度大幅下降，给航拍遥感场景下小目标检测带来巨大挑战。根据国际光学工程学会定义，尺寸小于原图的 0.12%可认为是小目标5。现阶段遥感领域的小目标检测性能的提升主要从多尺度融合和感受野角度出发。文献 6 对特征金字塔模块进行重构，添加跨层级横向连接以融合更多的通道特征，并在检测头前引入位置注意力（Coordinate Attention，CA）机制7以确保遥感小目标精确定位，虽然其对小目标具有较优的检测性能，但是采用直接去除顶层特征提取层的方式会对语义信息造成一定程度的损失，不利于应对复杂场景下的检测任务。文献 8 提出用于特征增强的特征图融合机制，利用卷积操作对不同深度、不同尺度的特征图深度以及尺度进行统一，融合得到检测能力更强的特征图，从而构建特征金字塔（Feature Pyramid Network，FPN）以增强小目标特征。尽管在遥感飞机和不同空中飞机数据集上的检测精度均有所提升，但是其检测速度无法满足实时性需求。文 献9基 于 Faster R-CNN10上 的 特 征 金 字塔11结构设计了特征门控模块和动态融合模块，依据不同尺度分配不同权重来区分目标尺度对特征融合的影响，解决在特征融合中共享同一权重的问题。但是，基于双阶段的目标检测算法本身在速度方面存在一定的局限性。此外，研究人员从感受野角度来提升小目标检测性能。文献 12 在 8 倍下采样后的特征映射中添加可变形卷积，对卷积中采样点的位置增加 1 个偏移，扩张实际感受野进而提升对小目标的识别精度，但是当目标存在背景遮挡时虚警率和漏检率较高。文献 13 参考 RFB（Receptive Field Block）14结构，采用多分支处理和空洞卷积设计特征增强模块以加强特征语义信息，减少小目标的检测精度损失，但是在遥感图像检测精度上仍有进一步提升的空间。文献 15 利用混合空洞卷积的方式提取特征，有效增 大 感 受 野，并 直 接 引 入 空 洞 空 间 金 字 塔 池 化（ASPP）16进行多尺度特征融合，以捕获更完整的卫星道路图像信息。但是以上方法使用场景单一，不能很好地应用在多场景任务中。因此，现有算法在小目标检测任务上取得了一定成效，但是无法有效权衡检测精度和速度，且未考虑卷积特征提取和下采样操作过程中的小目标信息传递丢失问题。本文提出基于 YOLOv5的遥感小目标检测算法YOLOv5-RS。通过构建轻量的并行混合注意力模块抑制图像中复杂背景和负样本的干扰，同时通过调整下采样倍数，保证传递更多的细节信息，并进一步设计卷积神经网络（Convolutional Neural Network，CNN）与 Transformer17相 结 合 的 特 征 提 取 模 块C3BT，将从特征图中获取到的局部与全局信息输出作为融合的底层特征。将原网络中的 CIoU18损失函数替换为 EIoU19损失函数，有效减少目标检测框的重叠，精准定位小目标的位置，从而提升模型对遥感小目标的检测性能。1YOLOv5-RS方法 近年来，目标检测领域越来越注重算法的工程化应用，对实时性要求大幅提高。考虑到一阶段网络 YOLOv5s20能兼顾检测精度与速度的特性，本文将其作为基础框架，从注意力模块、特征金字塔融合、损失函数 3个角度进行优化。在特征提取阶段，为减少小目标信息的丢失，本文使用 16倍下采样操作代替 32倍下采样，并添加浅层分支。上述操作虽然能够传递更多的浅层信息，但是缩小了感受野。基于此，本文构建具有更大感受野和更强表征能力的 C3BT 模块替换 SPPF 模块前的原 C3 特征提取模块，C3BT 由 CNN 和多头自注意力（Multi-Head Self-Attention，MHSA）组合而成，同时，将并行混合注意力模块 CBAM-P 嵌入到 FPN 结构中进行 Concat 融合浅层特征之前。从模型预测中发现，在小目标的定位任务中会存在大量重复的检测框，采用 EIoU 损失 函 数 具 化 预 测 框 与 真 实 框 之 间 的 长 宽 关 系。YOLOv5-RS网络整体结构如图 1所示。2572023年 9月 15日Computer Engineering 计算机工程1.1并行混合注意力模块注意力机制的本质是仿照人类视觉处理图像的过程，对特征图的不同位置予以各自的权重来表示不同的关注度。遥感图像往往包含复杂的背景，经过卷积层特征提取后存在待检测小目标信息占比少、背景以及背景中非检测物体信息占比多的情况，这些非感兴趣区域信息会对小目标检测产生干扰。为关注图像中待检测的小目标以及忽略无关的物体信息，本文借鉴了 CBAM（Convolutional Block Attention Module）21的 通 道 注 意 力 模 块（Channel Attention Module，CAM）和空间注意力模块（Spatial Attention Module，SAM）结构，提出通道注意力模块CAM-P和空间注意力模块 SAM-P，最终通过并行连接构成 CBAM-P 模块，使其分别沿通道和空间维度对 特 征 图 信 息 进 行 编 码。与 CBAM 模 块 相 比，CBAM-P 在 减 少 计 算 量 的 同 时 提 高 检 测 精 度。CBAM-P模块结构如图 2所示。首先对通道注意力模块进行改进，CAM-P 是基于 SENet22模块进行改进，将 SENet模块中全局平均池化（Global Average Pooling，GAP）后的全连接层替换为 33卷积层，并获取局部信息。因为小目标本身的感受野较小，所以全连接层对整张图片进行降维，导致待检测的小目标被淹没在与其他背景的平均特征之中。CAM-P特征图计算过程如式（1）所示：WC=(f1 1(f3 3(GGAP(X)（1）其中：表示 Sigmoid 激活函数；f1 1表示大小为 11的卷积；f3 3表示大小为 33 的卷积。输入特征 XH W C通过 GAP 后输出 11C 的一维通道注意力图，再通过激活函数重建得到权重特征图WC。然后对空间注意力模块进行改进，SAM-P 在原SAM 模块的基础上移除所有池化层，仅由 11 卷积层生成。该操作是考虑到池化会丢失小目标的关键特征而不利于检测。SAM-P特征图的计算过程是将输入特征 X 经 111 卷积的结果输入 Sigmoid 激活函数，得到空间注意力特征图WS，如式（2）所示：WS=(f1 1(X)（2）最终，将通道特征图WC与空间特征图WS分别与输入特征 X 逐元素相乘，采用逐元素相加方式将相乘后所得的特征图并行连接，输出特征XCS。上述过程的计算式如式（3）所示：XCS=X WCXWS（3）其中：表示逐元素相乘；表示逐元素相加。1.2特征金字塔改进特征金字塔网络结构如图 3 所示。YOLOv5 特征 图 的 融 合 部 分 采 用 路 径 聚 合 网 络（Path 图 1YOLOv5-RS网络整体架构Fig.1Overall architecture of YOLOv5-RS network图 2CBAM-P模块结构Fig.2Structure of CBAM-P module258第 49卷 第 9期李嘉新，侯进，盛博莹，等：基于改进 YOLOv5的遥感小目标检测网络Aggregation Network，PANet）23，其结构如图 3（a）所示。将主干网络经 8 倍、16 倍、32 倍下采样后输出P3，P4，P5 特征，分别与 FPN 自底向上的特征图进行融合。这种将浅层丰富细节信息和深层高语义信息融合的操作有利于多尺度目标检测，但是对于小目标，经过多次下采样后特征图所含小目标的有效特征信息较少，导致小目标检测精度降低。因此，本文对特征金字塔进行改进，保证传递更多的小目标细节信息并输出对小目标表征能力更强的特征图。改进后的特征金字塔PANet_RS结构如图3（b）所示。原始输入图像在经卷积下采样逐步映射为不同尺度特征图的过程中，所包含的有效像素信息逐渐减少。表1所示为将原始图像映射到不同特征层后所占像素的情况，P2、P3、P4、P5分别进行 4、8、16、32倍下采样。结合实际数据目标尺度分布情况，将小目标进一步划分为检测尺度小于1616像素的微小目标。从表 1 可以看出，小目标特征映射到 P5 特征层后所包含的小目标分辨率为 11 像素，微小目标的分辨率甚至不到 11 像素，对遥感小目标的检测效果甚微。考虑到直接裁剪 P5 检测层和其对应的分支，会造成深层网络间的语义信息缺失、感受野减小，进而影响分类任务的准确率。为缓解语义信息与位置信息之间的矛盾，本文在主干网络中去除32 倍的下采样层，使用 16 倍的卷积层进行替换，有效缓解连续下采样存在信息损失的问题。与 P3，P4，P5 相比，在 P2大尺寸检测层中含有丰富的纹理和轮廓信息，有利于图像中小目标的检测，因此添加 P2检测头并使用 CBAM-P注意力模块来抑制 P2特征层的噪声信息。为进一步平衡速度与精度，本文仍然保持3个尺度的检测头预测结构，但调整检测头为 P2，P3，P4 所对应的检测分支。虽 然 上 述 通 过 11 卷 积 替 代 用 于 下 采 样 的33卷积操作可以保证传递更多信息，但是 11卷积操作与YOLOv5网络相比感受野减少，导致图像中大尺寸目标的检测精度下降，整体精度提高不明显。基于此，本文构建基于 CNN 与 Transformer相结合的模块 C3BT，提取表征能力更强的特征图作为 FPN的底层特征。具体操作是将 YOLOv5特征提取基础单元Bottleneck 中的 33 卷积替换为 BottleTransformer24中的MHSA。BottleTransformer结构如图4所示。MHSA 层由多个 Self-Attention 模块组成，每个Self-Attention 模块在不同空间中捕获全局特征信息，最终将每个 Self-Attention 模块输出的信息进行拼 接 形 成 带 注 意 力 权 重 的 新 特 征 图 Z。每 个Self-Attention 模块的具体操作过程如式（4）所示，q（query）、k（key）、v（value）如式（5）所示：AAttention(qkv)=Softmax(qrT+qkTdk)v（4）q=XWQk=XWKv=XWV（5）其中：dk为缩放因子；r 为相对位置编码；q、k、v 分别由输入矩阵 X 与 3个初始化矩阵WQ、WK、WV相乘所得。从图 4 可以看出，MHSA 是从整张特征图中捕获信息，将全局信息间的依赖关系与 CNN 局部信息关系相结合，构成参数量更少且信息建模能力更强的特征图。1.3EIoU损失函数原始 YOLOv5 采用 CIoU 计算定位损失，其在IoU（Intersection over Union）损失的基础上，添加边界框回归的重叠面积、中心点距离及边长纵横比作为惩罚项因子。CIoU 及其惩罚项计算式如式（6）和式（7）所示：CCIoU=IIoU-2()bbgtc2-v（6）RCIoU=2()bbgtc2+v（7）表 1原始图像映射到不同特征层后的分辨率 Table 1Resolution of original image mapping to different feature layers 单位：像素原始图像分辨率161632329696映射后图像分辨率P2特征层44882424P3特征层22441212P4特征层112266P5特征层0.50.51133图 4BottleTransformer模块结构Fig.4Structure of BottleTransformer module图 3特征金字塔网络结构Fig.3Structure of feature pyramid network2592023年 9月 15日Computer Engineering 计算机工程其中：b、bgt分别表示预测框和真实框的中心点；表示 2个中心点间的欧氏距离；c表示包含预测框和真实框的最小闭包区域形成的对角线距离；v用来测量宽高比差异。本文采用 EIoU 损失函数计算定位损失，将宽高纵横比的损失项拆分，分别计算宽和高的差异值，计算式如下：LEIoU=1-IIoU(AB)+2()bbgt()hc2+()wc2+2()hhgthc+2()wwgtwc （8）其中：w、wgt、h、hgt分别表示预测框和真实框的宽度及高度。在同一目标区域内采用 CIoU 和 EIoU 损失函数的预测结果如图 5所示。CIoU 损失增加了宽高比的一致性，使得检测框回归过程更加稳定，收敛的精度更高。但是参数 v只反映了宽高比间的相对差异，而不是 w 与wgt、h与hgt之间的真实关系。CIoU 定位损失如图 5（a）所示。CIoU 损失函数生成的预测框与真实框相比，其宽边差异很大且无法精确定位。利用预测框与标注框宽高之间的真实差监督反向传播过程，得到损失函数最优解。该过程提高回归精度进而提升小目标检测性能。EIoU定位损失如图 5（b）所示，EIoU 定位损失函数生成的预测框高度和宽度与真实框相似。2实验结果与分析 2.1实验配置实验环境：Ubuntu20.04操作系统，显卡 GeForce RTX 3090，训 练 平 台 Python3.8，深 度 学 习 框 架PyTorch。模型训练参数设置：迭代次数为 200，批大小 batch_size为 16，其余参数均为默认值。2.2数据预处理本 文 采 用 武 汉 大 学 发 布 的 航 空 遥 感 影 像DOTAv1.5 数据集25对所提小目标检测算法进行评估，选 取 小 型 车 辆（Small Vehicle，SV）、大 型 车 辆（Large Vehicle，LV）和轮船（SH）3种小目标数据集作为原始图像，图像大小一般为 8003 000 像素不等。若将小目标数据集图像直接送入检测网络训练，其分辨率过大，易造成显存溢出。若等比例缩放会直接导致图像过度压缩而丢失大量小目标细节信息。因此，本文使用图像裁剪方式对数据集进行预处理，将样本统一裁剪为 1 0241 024像素的图片，在切割图像时保留一定重叠区域（gap）来防止切割边缘目标信息丢失，本文将 gap设置为 200。将处理后得到的 9 240 张图像构成 DOTA-v 数据集，共计标注 457 528 个实例，平均单幅图像包含49.5 个待测实例，按照 6 2 2 比例划分为训练集、验证集和测试集。图 6 所示为 3 类目标的标注框占原图像尺寸的比例分布。从分布情况可以看出，SV类目标几乎全部为占比小于 0.1的小目标，LV 和 SH类目标也大多集中在 0.1 范围内。在 COCO 数据集26中大、中、小目标占比约为 4 3 2，而 DOTA-v数据集为 7 2 1，显然 DOTA-v数据集的小目标占比更高，可用于验证本文算法的合理性。2.3评价指标本文选取目标检测领域 3 种常用的评价指标对算法进行定量评价。1）精确度（P）和召回率（R）的计算式如式（9）和式（10）所示27：P=TTPTTP+FFP（9）R=TTPTTP+FFN（10）其中：TTP（True Positive）表示图像中待检测目标被正图 5同一目标区域预测结果Fig.5Prediction results for the same target area图 6在 DOTA-v数据集中各类目标尺寸分布情况Fig.6Distribution of various objects size on DOTA-v dataset260第 49卷 第 9期李嘉新，侯进，盛博莹，等：基于改进 YOLOv5的遥感小目标检测网络确识别且 IoU大于阈值；FFP（False Positive）表示检测目 标 未 被 正 确 识 别 且 IoU 小 于 阈 值；FFN（False Negative）表示没有被检测到的目标。2）平均精度（mAP）计算式如式（11）所示：mmAP=1N0N01Pn(r)dr（11）其中：N 代表数据集中检测目标的类别个数；Pn表示某一类别的 AP 值。mAP0.5 和 mAP0.5 0.95 常用来评估模型性能，mAP0.5关注模型精确率随召回率变化趋势，mAP0.5 0.95 更关注模型在不同IoU阈值下的综合表现，反映检测框与真实框的拟合程度。本文若无特殊说明，mAP默认为 mAP0.5。3）帧速（Frames Per Second，FPS）28 表示模型每秒能处理图片的数量，单位为帧/s，通常用于衡量模型的实时性，辅以参数量和计算量（GFLOPs）综合判别模型性能。2.4结果分析2.4.1注意力机制改进实验为验证本文所提 CBAM-P 模块的有效性，本文进行注意力机制对比实验，比较参数量和 mAP 这2 个指标，其中，CBAM-S 表示 CAM-P 与 SAM-P 的串行模块，CBAM-P表示 CAM-P与 SAM-P的并行模块。CBAM-P模块对模型性能影响如表 2所示。CBAM 首先使用 CAM 给不同通道分配不同的权重，然后使用 SAM 进行空间信息校准。从表 2 可以看出，在 YOLOv5s中直接嵌入 CBAM 模块后相比YOLOv5s mAP 指标仅提升 0.1 个百分点，然而在YOLOv5s中直接嵌入 CBAM-P模块后与 YOLOv5s+CBAM 相 比 不 仅 参 数 量 减 少，而 且 mAP 提 高 了0.3 个百分点。分析认为有 2 个方面的原因：1）在小目标信息本身不足的情况下，池化操作和全连接操作带来的负面影响大于正面影响，替换全连接层和移除池化层后的模块后所得的特征图更有利于小目标检测；2）得益于并行的连接结构，在串行结构中，空间注意力机制的输入是经过通道注意力模块后所得，使得目标浅层信息再次减少，即使具有更多的语义信息也无法对小目标进行定位，相反可能导致目标错检。为进一步直观对比 CBAM 模块与 CBAM-P 模块的效果，本文采用 Grad-CAM（Gradient-weighted Class Activation Map）29热力图进行可视化，可视化结果如图 7 所示（彩色效果见 计算机工程 官网HTML 版）。图 7（a）中标注框表示将地面停车线误判为感兴趣目标，从图 7（b）可以看出，采用 CBAM-P结构后的热力图没有出现误判现象。该实验结果进一步说明注意力机制的减法操作和并行结构能够有效提高空间维度特征的提取能力，更好地区分前景和背景信息，使得关注区域与检测目标区间更紧凑，证明 CBAM-P模块对小目标检测是有效的。2.4.2特征融合层改进实验检测头分支数量与性能关系如表 3 所示。本文综合考虑参数量与检测精度，选取P2，P3，P4对应的预测分支输出检测头。从表 3 可以看出，采用减少下采样倍数并添加P2检测头模块后的结构 P2，P3，P4，P5，与 P2，P3，P4 检测头相比 mAP 提高约 0.2个百分点，证明该结构能够传递更多的小目标信息。但受限于降采样操作所造成感受野缩小和浅层特征 P2中的噪声干扰，mAP 提升不是很明显。基于此，本文进一步验证C3BT 模块的有效性，选取P2，P3，P4检测头进行对照实验，并增加各类目标 mAP 指标以直观反映其对不同尺寸目标检测性能的影响，对比结果如表 4所 示，“”表 示 在 基 准 模 型 YOLOv5s 中 添 加 该模块。表 2CBAM-P模块对模型性能影响 Table 2Influence of CBAM-P module on model performance模型YOLOv5sYOLOv5s+CBAMYOLOv5s+CBAM-SYOLOv5s+CBAM-P参数量7 018 2167 029 0627 018 6067 018 606mAP/%81.681.781.882.0图 7注意力机制可视化结果Fig.7Visualization results of attention mechanism表 3检测头分支数量与模型性能关系 Table 3Number of detection head branches in relation to model performance检测头P2,P3,P4P3,P4,P5P2,P4,P5P2,P3,P4,P5参数量4 338 0565 597 9765 585 8325 599 136mAP/%81.881.682.082.02612023年 9月 15日Computer Engineering 计算机工程从表 4可以看出，为传递更多小目标而采用降采样操作，添加P2，P3，P4 结构后，相比 YOLOv5s 的SV目标的 mAP提高 1.2个百分点，但 LV类目标 mAP下降 0.4 个百分点，SH 类目标 mAP 下降 0.3 个百分点。感受野的缩小影响了 LV 类和 SH 类中大尺寸和中尺寸目标，无法捕捉到全部的特征。在添加了C3BT 模块后，与 YOLOv5s 相比，mAP 提升了 0.8 个百分点，其中SV类目标mAP提升了2.2个百分点，SH类目标 mAP提升 0.1个百分点，说明 C3BT模块能有效地提取更具判别能力的特征用于检测小目标，同时扩张了感受野保证大尺寸和中尺寸目标的精度。2.4.3模块消融实验为验证各模块的有效性，本文在遥感数据集DOTA-v上进行消融实验。本文在 YOLOv5s基础上依次添加 CBAM-P、特征层改进模块（PANet_RS）以及 EIoU损失函数。消融实验结果如表 5所示。从 表 5 可 以 看 出，本 文 所 提 CBAM-P、PANet_RS、EIoU损失函数对模型性能具有一定的有效性，YOLOv5s+CBAM-P+PANet_RS+EIoU 模型大小缩小为 10.1 MB，整体 mAP 与 YOLOv5s相比提高了 1.5个百分点。2.5对比实验2.5.1预测结果分析从上述实验中可以发现，本文在 DOTA-v 数据集上的整体预测效果优于 YOLOv5s。为了进一步反映改进模型的性能，在所有预测类上的 mAP 变化结果如表 6 所示，并通过遥感图像进行主观视觉定量评价。从 表 6 可 以 看 出，与 YOLOv5s 相 比，YOLOv5-RS 在 SV、LV 和 SH 3 个类别目标的 mAP均得到提高，尤其是 SV 的检测精度提升了 3.2 个百分点。SH 类小目标大多呈现密集分布，水平检测框之间的重合导致精度提升不明显。YOLOv5-RS 与 YOLOv5s 检测效果对比如图 8所示（彩色效果见 计算机工程 官网 HTML版）。从图 8 可以看出：第 1 组 YOLOv5-RS 算法对小目标定位更加准确，预测框能更好地定位到物体所在的位置，且减少了预测框重叠的现象；第 2组在遮挡场景下，YOLOv5-RS算法可以更准确地检测被树遮挡的车辆；第 3组在密集场景下，YOLOv5-RS算法可以检测出更多的船类目标，在目标识别与定位能力上均优于 YOLOv5s算法。2.5.2不同算法对比实验本 文 以 一 阶 段 EfficientDet30、YOLOx31、TPH-YOLOv532、YOLOv733以及 YOLOv5s 5 种算法作为对照组进行实验，包括近年来在小目标检测领域的常用框架和最新的改进框架。EfficientDet系列网络以权衡速度与精度为前提，实现对双向特征融合结构的动态调整。本文选取兼顾速度与精度的图 8YOLOv5-RS与 YOLOv5s算法的检测效果对比Fig.8Detection effects comparison between YOLOv5-RS and YOLOv5s algorithms表 5消融实验结果 Table 5Ablation experiment resultsYOLOv5sCBAM-PPANet_RSEIoUmAP/%81.682.082.482.283.1模型大小/MB14.1014.109.9314.1010.10表 6YOLOv5s与 YOLOv5-RS各类别 mAP对比 Table 6Comparison between YOLOv5s and YOLOv5-RS by category mAP%模型YOLOv5sYOLOv5-RSmAPSV73.576.7LV82.583.6SH88.889.0表 4C3BT模块对模型性能影响 Table 4Influence of C3BT module on model performance%C3BTP2,P3,P4mAPSV73.574.775.7LV82.582.182.5SH88.888.588.9平均值81.681.882.4262第 49卷 第 9期李嘉新，侯进，盛博莹，等：基于改进 YOLOv5的遥感小目标检测网络EfficientDet-d4 作为对照组。YOLOx 采用无锚框方式，并 通 过 解 耦 YOLO Head 提 升 网 络 的 性 能。TPH-YOLOv5 是 基 于 YOLOv5s 进 行 改 进。由 于TPH-YOLOv5 集成 CBAM、Transformer 等模块，因此在无人机小目标检测数据集上性能表现突出。YOLOv7 设计新的内部组件模块，结合新的标签分配策略使其架构在速度和精度上均取得较优的表现。不同算法的评价指标对比如表 7 所示，加粗表示最优数据。与上述 5 种算法相比，YOLOv5-RS 在mAP0.5、mAP0.5 0.95和模型大小 3种评价指标上取得最优结果，在精确率与召回率的平衡、检测框与预测框的拟合方面表现突出。YOLOv5-RS 检测速度为 65.4 帧/s，其兼顾精度与实时性，对小目标检测性能更优。3结束语 为解决现有算法的小目标检测精度远低于大、中目标的问题，本文提出一种基于并行注意力机制和融合更多低层级特征的检测算法 YOLOv5-RS。在特征金字塔融合结构中添加 1个浅层特征分支以反馈更多的浅层特征，并利用所提的注意力模块抑制分支中的噪声干扰。通过调整下采样倍数，并融合全局和局部信息生成特征提取能力更强的特征提取模块。在预测阶段，采用 EIoU 损失函数有效缩小真实框与预测框之间的差异。实验结果表明，相比现有一阶段算法，本文算法具有较优的小目标识别能 力，在 复 杂 场 景 下 鲁 棒 性 较 优。下 一 步 将 在一阶段网络中引入旋转检测方法，解决目标方向不确定所存在边界框重叠问题。参考文献 1 ORFANUS D，DE FREITAS E P，ELIASSEN F.Self-organization as a supporting paradigm for military UAV relay networks J.IEEE Commun