重庆城乡规划和建设局网站,不合理的网站,简单的模板网站,亚马逊新店投广告是免费的吗目录
论文信息
论文创新点
1.自适应轻量通道分割和洗牌#xff08;ALSS#xff09;模块
2.轻量坐标注意力#xff08;LCA#xff09;模块
3.单通道聚焦模块
4.FineSIOU损失函数
摘要
架构设计
轻量高效网络架构 - ALSS模块
LCA模块
单通道聚焦模块
损失函数优…目录
论文信息
论文创新点
1.自适应轻量通道分割和洗牌ALSS模块
2.轻量坐标注意力LCA模块
3.单通道聚焦模块
4.FineSIOU损失函数
摘要
架构设计
轻量高效网络架构 - ALSS模块
LCA模块
单通道聚焦模块
损失函数优化 - FineSIOU
实验与分析
消融实验
对比实验
Coovally AI模型训练与应用平台 论文信息 题目ALSS-YOLO: An Adaptive Lightweight Channel Split and Shuffling Network for TIR Wildlife Detection in UAV Imagery ALSS-YOLO用于无人机红外图像中野生动物检测的自适应轻量通道分割和洗牌网络 作者Ang HeXiaobo LiXimei WuChengyue SuJing ChenSheng XuXiaobin Guo 论文链接: https://arxiv.org/abs/2409.06259 论文创新点
1.自适应轻量通道分割和洗牌ALSS模块
作者提出了一个新颖的ALSS模块该模块采用自适应通道分割策略来优化特征提取并集成了通道洗牌机制以增强通道间信息交换。这一改进提高了对模糊目标的检测准确性尤其是在处理由抖动引起的模糊和重叠目标时。
2.轻量坐标注意力LCA模块
开发了一个LCA模块该模块使用自适应池化和分组卷积来整合跨维度的特征信息。这一模块在保持轻量操作的同时确保了高检测精度和对抖动及目标重叠的鲁棒性。
3.单通道聚焦模块
为了提高红外图像的特征表示效率作者设计了一个单通道聚焦模块将每个通道的宽度和高度信息聚合到四维通道融合中从而改善了特征的表示和提取。
4.FineSIOU损失函数
作者修改了定位损失函数提出了FineSIOU该函数通过强调与小目标相关的损失值提高了小目标的定位精度。此外FineSIOU将角度成本作为独立项纳入总成本的计算中加速了模型的收敛速度。 摘要
搭载热红外TIR相机的无人机在打击夜间野生动物偷猎中发挥着关键作用。然而TIR图像常面临抖动和野生动物重叠等挑战需要无人机具备识别模糊和重叠小目标的能力。当前部署在无人机上的传统轻量网络难以从模糊小目标中提取特征。为了解决这一问题作者开发了ALSS-YOLO这是一种针对TIR航空图像优化的高效轻量检测器。首先作者提出了一种新颖的自适应轻量通道分割和洗牌ALSS模块。该模块采用自适应通道分割策略来优化特征提取并集成了通道洗牌机制以增强通道间信息交换。这改善了模糊特征的提取对于处理抖动引起的模糊和重叠目标至关重要。其次作者开发了一个轻量坐标注意力LCA模块该模块采用自适应池化和分组卷积来整合跨维度的特征信息。该模块确保了轻量操作同时保持了高检测精度和对抖动和目标重叠的鲁棒性。此外作者开发了一个单通道聚焦模块将每个通道的宽度和高度信息聚合到四维通道融合中提高了红外图像的特征表示效率。最后作者修改了定位损失函数强调与小目标相关的损失值以提高定位精度。在BIRDSAI和ISOD TIR UAV野生动物数据集上的广泛实验表明ALSS-YOLO实现了最先进的性能。 架构设计
作者的ALSS-YOLO检测器的整体架构源自最新的YOLOv8-n目标检测器。作者引入了一个创新的ALSS模块作为主干网络。通过利用通道分割和随后的独立处理网络可以学习输入数据的各种特征提高模型识别复杂或微妙特征的能力。利用瓶颈设计和深度卷积ALSS模块实现了高效的特征提取参数数量最少。此外通过通道洗牌和跨不同路径的特征融合模型的表达能力和泛化能力得到了显著提升。其次受通道注意力CA[46]的启发作者引入了一个LCA模块该模块对输入特征图的高度和宽度维度应用自适应平均池化从而生成跨两个空间维度的全局上下文特征。解耦和独立处理每个方向的特征提高了模型捕获全局信息的能力优化了计算效率和参数使用。此外特别为单通道UAV FIR图像设计了一个单通道聚焦模块。该模块将单通道的宽度和高度信息聚合到四维通道中从而在较小的空间分辨率下保留更多细节。宽度和高度数据通过卷积操作与通道信息集成从而提高了特征表示的效率和准确性。最后引入了FineSIOU它强调了形状成本与真实边界框尺寸之间的相关性并将角度成本作为独立项纳入总成本的计算中从而提高了回归速度和小目标检测能力。图2展示了ALSS-YOLO检测器的架构。 轻量高效网络架构 - ALSS模块
在深度学习中轻量和效率是当前研究的热点之一。设计一个轻量高效的神经网络架构对于资源受限的场景至关重要。在这种情况下作者引入了一个名为ALSS的新型网络架构旨在实现轻量高效的目标检测。该模块的结构如图3所示。 此外图4展示了ALSS模块作为下采样操作时的配置。在较低的特征级别重点是捕获细粒度细节和低级特征我们设置一个较低的α值。这确保了只有一小部分通道通过图3的A部分进行处理通常涉及基本的卷积操作。大多数通道被导向更复杂的多级网络图3的B部分旨在增强特征提取和捕获复杂模式。随着特征级别的提高对抽象和高阶语义信息的需求变得更加关键。为了适应这一点我们在更高级别增加α值。这种调整将输入通道直接连接到输出通道促进了复杂特征的提取和网络的表示能力。通过优先考虑低级别的卷积操作并将重点转向更高级别的ResNet类结构进行残差学习这种策略平衡了计算效率和特征丰富性之间的权衡优化了网络在不同特征层次的性能。 此外在低级别特征中网络结构如图3的A部分所示我们使用卷积操作来提取特征。这是因为在浅层卷积层中特征更加局部化和详细。通过卷积操作网络可以获得多种过滤器来检测图像中的局部特征包括边缘、纹理和其他特征。这种方法使网络能够高效地提取基本特征并逐步将它们融合成更高级的特征表示。相反在高级别特征中我们更倾向于使用恒等连接。在深层卷积层中特征变得更加抽象和语义化包含更高级别的语义信息如物体形状、类别等。通过使用恒等连接我们可以保留这些高级特征并防止网络学习到的重要信息被过度压缩或丢失。这个过程可以表示为这种自上而下的信息传播促进了信息的流动和共享最终提升了模型的性能和泛化能力。参数设置策略的α和网络结构选择策略的合理性将在后续的实验部分得到验证。为了减少模型的参数数量并提高计算效率我们使用瓶颈结构和深度卷积图3的B部分。瓶颈结构通过在较低维度提取特征有效地降低了计算成本而深度卷积进一步提高了网络的感知范围和特征提取能力特别适合处理更复杂的语义信息。具体来说首先应用一个3×3的卷积核步长为1以实现降维。这种操作引入了一个降维系数β它修改了输出通道的数量以调节模型的参数数量和计算工作量。在实践中当处理具有更多通道的级别时选择较大的β值以通过更积极地压缩通道来显著降低资源消耗。这使得模型能够有效地管理增加的计算负载和内存使用。相反在通道数量较低的级别使用较小的β值以确保模型能够保持足够的特征提取能力保留特征图中的关键信息。因此β的选择是在资源效率和捕捉及表示关键特征的能力之间的平衡。深度卷积的使用进一步补充了这一策略通过减少参数数量和计算复杂性同时保持数据中的空间和通道关系为轻量而有效的架构做出了贡献。在降维操作之后为了进一步增强模型的特征提取能力这种策略然后应用了一层3×3的深度卷积。深度卷积通过单独对输入特征图的每个通道进行卷积来增强网络的非线性从而减少了与传统卷积相比的参数数量然后为了调整模型的输出维度再次应用一个3×3的卷积核并将输出维度设置为Cout - αC 其中Cout代表输出通道的数量α表示前面提到的预定义缩放因子。作者的网络设计还遵循了ShuffleNetV2[42]中提到的轻量网络设计原则避免过多的分支结构选择沿通道方向连接特征图以减少计算负担。这种设计不仅提高了网络的计算效率还增强了信息交换和特征表示能力使网络更适合各种复杂场景中的实用应用。根据文献[47]的建议在深度卷积之后不使用SiLU激活函数。在模块的最后阶段采用了通道洗牌操作来增强不同特征通道之间的信息交换。这种操作重新排列了输入特征图的通道使得之前隔离的特征分支能够共享学习到的信息从而改善了信息流和特征表达的多样性。整个网络架构在保持轻量设计的同时更好地捕获和整合多尺度、多角度的特征信息增强了从复杂场景中提取特征的能力。在作为下采样结构的ALSS模块中作者设计了三个模块来下采样αCin输入特征图通道。如图4的A部分所示根据网络模型内的特征级别从低到高依次应用以下操作卷积、卷积-池化和池化这些操作能够在低级别实现详细的特征提取并帮助捕获图像的基本纹理和形状信息。在更高级别纳入了类似于ResNet的残差学习结构以促进网络的有效训练特别是在网络非常深的情况下。这种方法使网络能够更好地学习高级抽象特征而不会丢失低级信息。对于(1 - α)Cin输入特征图通道瓶颈阶段的初始3×3卷积中使用了2的步长来减少特征图的宽度和高度在ALSS-YOLO网络架构中从下到上的α系数设置为(0.4, 0.4, 0.5, 0.6, 0.7, 0.2, 0.3, 0.2, 0.2)而β系数定义为(0.4, 0.4, 0.5, 0.6, 0.6, 0.8, 0.8, 0.8, 0.8)。 LCA模块
最近深度学习的进步强调了注意力机制在提升神经网络性能方面的重要性特别是在各种任务中优化特征表示。具体来说引入的CA[48]机制进一步优化了这一过程。它不仅关注特征的通道维度还强调了特征空间的坐标信息。这种设计使模型能够更有效地整合全局上下文信息同时解析空间维度。为了增强特征提取的效率并最小化参数数量本文提出了一种改进的坐标注意力机制即LCA。图5提供了CA和LCA之间的比较视图而图6揭示了LCA的内部结构细节。在以下部分中将详细阐述LCA的设计原则和实现细节。对于输入张量X其尺寸为C×W×H其中H、W和C分别代表高度、宽度和通道数我们采用了一种特殊的方法来捕获沿特定空间维度的特征。这是通过应用两个独立的池化操作实现的每个操作使用不同的内核尺寸一个沿水平维度使用(H, 1)的池化内核另一个沿垂直维度使用(1, W)的池化内核。这些操作分别在每个通道上执行以保留沿各自维度的空间信息。对于输入张量X的第c个通道在高度h处应用水平池化内核尺寸为(H, 1)而在宽度w处应用垂直池化内核尺寸为(1, W)。 这两个方向的池化操作使得捕获沿两个空间维度的特征成为可能从而产生了一个方向感知的特征图增强了网络识别精确位置和提高空间意识的能力。这种新颖的方法将沿两个维度的特征处理解耦允许高效且成本效益的特征提取特别是对于计算资源有限的移动设备来说非常有利。一旦通过上述池化操作生成了特征图它们就会经历不同的1×1深度可分离卷积。这些卷积促进了通道间的线性变换同时结合非线性激活函数来调节注意力权重。LCA模块进一步完善了网络聚焦空间特征的能力。通过沿高度和宽度维度应用自适应平均池化然后进行深度可分离卷积该模块捕获了全局空间信息同时保持了计算效率。通过这些过程生成的注意力权重调节输入张量增强了网络的区分能力。在将LCA模块集成到网络架构时特别关注具有较高分辨率特征图的层尤其是对于检测小物体。根据作者的分析小物体在这些特征图上更有效地被检测。因此LCA模块被策略性地放置在第15层靠近检测头以最大化在复杂场景中小物体的检测精度。作者将在以下实验部分验证其合理性。总之LCA模块结合自适应池化和深度可分离卷积提供了一种强大的机制来增强空间注意力同时保持低计算复杂性。其在网络架构中的策略性集成显著提高了特征区分和检测能力特别是在以复杂空间结构和存在小、重叠物体为特征的环境中使其成为需要精确度和效率的任务的基本组成部分。
LCA代码如下
class LCA(nn.Module):def __init__(self, input_channel, reduction32):super(LCA, self).__init__()self.pool_h nn.AdaptiveAvgPool2d((None, 1))self.pool_w nn.AdaptiveAvgPool2d((1, None))self.conv1 nn.Conv2d(input_channel, input_channel, kernel_size1, stride1, padding0)self.bn1 nn.BatchNorm2d(input_channel)self.act h_swish()self.conv_h nn.Conv2d(input_channel, input_channel, kernel_size1, stride1, padding0,groupsinput_channel)self.conv_w nn.Conv2d(input_channel, input_channel, kernel_size1, stride1, padding0,groupsinput_channel)def forward(self, x): # torch.Size([2, 32, 64, 64])identity xb, c, h, w x.size()x_h self.pool_h(x) # torch.Size([2, 32, 64, 1])x_w self.pool_w(x) # torch.Size([2, 32, 1, 64]) a_h self.conv_h(x_h).sigmoid() # torch.Size([2, 32, 64, 1])a_w self.conv_w(x_w).sigmoid() # torch.Size([2, 32, 1, 64])out identity * a_w * a_h # torch.Size([2, 32, 64, 64])return out 单通道聚焦模块
一个高效且准确的特征提取机制可以显著提升模型的性能。对于单通道TIR图像传统的特征提取方法面临着信息利用不足或高计算资源消耗的挑战。 为此作者提出了一个单通道聚焦模块旨在通过将宽度和高度信息有效地聚合到通道维度并通过优化的卷积策略增强模型的特征表示能力。对于给定的x ∈ RN×1×H×W TIR图像其中N是样本数批量大小1是通道数H和W分别是高度和宽度。它首先在宽度和高度维度上被分割并将分割的区域在通道维度上组合生成一个x ∈ RN×4×H/2×W/2图像如方程(13)所示。每个通道携带原始TIR图像中的局部信息新的多通道组合在通道维度上融合了这些局部信息从而实现了信息聚合的效果。随后生成的特征图通过一个卷积核尺寸为6、步长为2的卷积操作进行处理参考方程(15)这扩展了全局特征提取的感受野有助于从输入特征图中更有效地捕获空间信息。这进一步帮助模型学习更全局和抽象的特征提高了识别复杂模式和结构的能力并将特征图的大小减半。这种方法有助于减少后续层的参数数量和计算负载同时保留关键特征信息。 class Focus(nn.Module):Focus wh information into c-space.def __init__(self, c1, c2, k1, s1, pNone, g1, actTrue):Initializes Focus object with user defined channel, convolution, padding, group and activation values.super().__init__()self.conv Conv(c1 * 4, c2, k, s, p, g, actact)# self.contract Contract(gain2)def forward(self, x):Applies convolution to concatenated tensor and returns the output.Input shape is (b,c,w,h) and output shape is (b,4c,w/2,h/2).return self.conv(torch.cat((x[..., ::2, ::2], x[..., 1::2, ::2], x[..., ::2, 1::2], x[..., 1::2, 1::2]), 1))# return self.conv(self.contract(x)) 损失函数优化 - FineSIOU
在YOLOv8中损失函数包括分类损失和边界框回归损失。与YOLOv8中用于边界框回归损失的CIOU相比SIOU[49]损失函数考虑了真实框和预测框之间的角度。它使模型在训练期间更容易和更快地接近真实框显著提高了模型的训练效率和准确性尤其是在检测复杂背景和重叠场景中的对象时。在SIOU的基础上作者考虑了边界框回归与真实框大小之间的密切关系对于小目标的回归难度大于大目标通过调整形状损失来强调真实框的大小和形状显著提高了小目标检测的准确性。此外角度成本被单独作为一个项纳入总成本的计算中以加速收敛。这种增强的损失函数被称为FineSIOU。为了提供背景简要回顾一下SIOU是必要的它整合了四个组成部分角度成本、距离成本、形状成本和IOU成本。图8显示了SIOU计算方案。 SIOU将形状成本定义为[49]ω其中wgt和hgt分别代表真实框的宽度和高度。W和h分别代表预测框的宽度和高度。θ用于控制对形状损失的关注程度。其中IOU代表IOU损失∆代表距离成本Ω对应于形状成本角度成本λ被整合到距离成本的计算中。在作者对形状成本的计算进行研究时发现形状回归的难度与真实框的大小密切相关。如图9(a,b)所示在保持角度成本和距离成本不变的前提下SIOU得到的形状成本结果是相同的。相比之下作者认为图9(a)中小真实框与大预测框之间的回归难度应该高于图9(b)中大真实框与小预测框之间的回归难度。 此外在距离成本为零的特定情况下如图10(a,b)所示作者认为图10(a)的形状成本应该低于图10(b)。基于以上分析作者设计的形状成本如下ωω参数θ用于调节形状损失的重视程度。当θ被赋予1的值时形状优化过程将被优先考虑。作者的实验结果表明将其设置为6可以获得良好的结果。经过改进的形状成本更加关注真实框的大小特别是对于以小目标为主的航空图像。较小的边界框将产生更高的形状损失值这与小目标比大目标具有更大的回归挑战这一直观观点是一致的。表II提供了在各种真实框和预测框下计算的形状成本的例子。同时作者还关注角度成本并将其作为单独的一项纳入总成本的计算中。其中σ代表真实框中心点与预测框之间的欧几里得距离而bh表示真实框中心点与预测框之间的高度差。计算值0.9847对应于角度α5°。当角度小于5°时预测框与真实框之间的角度损失不再是主要关注点但距离损失和形状损失应该得到更多关注。η是一个用于调节角度损失重视程度的参数作者的实验表明将其设置为3可以获得良好的结果作者认为更精细的参数调整可能会带来更好的结果。 实验与分析 消融实验 为了评估本研究提出的增强策略的效果我们进行了系列消融实验以量化其影响。如表 III、表 IV 和图 13 所示每项技术增强均带来了性能提升。在 YOLOv8-n 基准模型的基础上为了保持模型参数量的大致一致性我们将模型的宽度超参数从 0.25 调整为 0.18并将其标记为 YOLOv8-N’尽管我们还尝试调整了深度超参数但效果不佳mAP0.50 仅达到 0.863。本研究提出了一种以 ALSS 模块为核心的网络结构同时引入单通道焦点模块。原始模型中第 16 至 19 层和第 20 至 23 层的步幅为 2 的 3 × 3 卷积被最大池化和 1 × 1 点卷积取代。此外还引入了 LCA 模块并相应修改了损失函数。同时我们进行了 CA 模块与 LCA 模块性能对比的对照实验。
为探究ALSS模块作为模型核心组件的有效性本研究对 YOLOv8-N’ 进行了改进将原始的 C2F 模块替换为 ALSS 模块并调整了不同特征尺度通向检测头的通道比例。新的网络结构命名为M1。根据表 III第 1 行和第 2 行中的数据与 YOLOv8-N’ 相比M1 模型在 mAP0.50 指标上提高了0.3 个百分点同时参数量减少了 17.4%。特别值得注意的是精确度显著提高从 0.869 增加到 0.889。从图 13(a) 可以看出M1 模型的收敛速度快于 YOLOv8-N’而图 13(c) 显示在高置信度阈值设置下YOLOv8-N’ 的 F1 分数显著下降这一现象在后续图 14 中进一步得到验证。这些结果充分证明了 ALSS 模块的轻量化和高效性。
此外为验证单通道焦点模块在增强模型识别复杂模式和结构能力方面的有效性本研究在 M1 模型的第一层中引入了该模块形成M2 网络结构。表 III第 2 行和第 3 行显示在参数量基本不变的情况下M2 模型的 mAP0.50 提高了 **1%**。同时表 IV第 2 行和第 3 行指出除未知类别的 AP 值略有下降外其他类别的 AP 值均有所提高。
为了进一步减少参数量和模型复杂度我们将 M2 模型中第 16 至 19 层和第 20 至 23 层的 3 × 3 卷积层步幅 2替换为步幅为 2 的最大池化层和点卷积结构称为M3。通过最大池化进行下采样提高了模型的空间不变性。相比 3 × 3 卷积核的卷积层点卷积的参数量和计算成本更低有助于调整输入特征图的深度以实现跨通道信息整合。根据表 III 和表 IV第 3 行和第 4 行的数据与 M2 相比M3 模型的参数量减少了 3.4%同时 mAP0.50 **仅降低了 0.1%**各类别的 AP 值也相似。图 13 还显示M3 模型在多个评价指标下维持了接近的性能。
此外为评估 LCA 模块在增强特征判别能力和整体网络性能方面的作用同时保持相对较低的复杂度和高计算效率本研究在 M3 模型的第 15 层集成了 LCA 模块从而形成M4 模型。从表 III 和表 IV第 4 行和第 5 行的对比来看尽管模型参数量增加了 1.3%mAP 提高了 0.3%。考虑到基准 mAP 值已达到 88.6%这一提升相当显著。同时召回率从 0.834 提升到 0.854各类别的 AP 值也各有增减。 损失函数的修改通常仅影响训练阶段对网络的推理时间没有影响。本文提出的 FineSIOU 损失函数是一种提高检测边界框精度的方法。从表 III 和表 IV第 5 行和第 6 行的数据以及图 13 的趋势可见引入新损失函数使 mAP0.50 提升了 **0.2%**并在模型收敛速度、PR 曲线和 F1 分数等方面表现最佳。模型实现了精度与速度的优化平衡在计算资源有限的无人机平台上实现了更高的精度。FineSIOU 损失函数中角度损失作为独立项的分离加速了收敛。鉴于 BIRDSAI 数据库中存在大量小目标模型对小目标检测的能力也得到了实验证明。
此外我们还进行了 CA 模块的对比实验如表 III 和表 IV第 6 行和第 7 行所示。实验表明在 BIRDSAI 数据集上LCA 模块的性能优于 CA 模块M5 模型的 mAP50 比 M6 高出 0.3%。在 ISOD 数据集[10]上也观察到了这一现象。这些发现表明LCA 模块在某些场景下具有增强性能的潜力。未来的工作将扩展这些评估至更多数据集以进一步验证性能差异并探索各模块在不同场景下的适用性。
如图 14 所示的消融实验基于置信度大于 0.6 的检测结果进行。从图 13(c) 可以看出当置信度为 0.6 时F1 值仅约为 0.1表明模型的召回率显著降低。如图 14(a) 所示有大量漏检但在我们改进的后续模型中漏检问题已被消除。此外ALSS-YOLO 检测框的整体置信度最高。图 15 展示了第 10、12、14 和 16 层模块检测区域的特征激活图可视化。这些可视化表明尤其是在集成 LCA 模块后网络能够有效关注被检测目标的整体即使在相对浅层网络也能聚焦于目标的语义信息展现出出色的特征抽象能力。这进一步凸显了 LCA 模块在提升特征表示和判别能力方面的作用从而有助于检测过程的整体性能提升。 对比实验
为了证明我们模型的优越性能我们精心调整了 ALSS-YOLO 框架的宽度和深度超参数衍生出两个新变种分别命名为 ALSS-YOLO-s 和 ALSS-YOLO-m。同时对 YOLOv8-n 模型的宽度超参数进行微调生成了 YOLOv8-n。如表 V 所示我们的 ALSS-YOLO-m 模型相比 YOLOv8-n 实现了 mAP0.50 提升 0.9%而 ALSS-YOLO-s 则比 YOLOv8-n 提高了 1.1% 的 mAP尽管两者参数量相近。值得注意的是尽管我们模型在检测“长颈鹿”类别时的 AP 略逊于 YOLOv8但在其他类别的错误检测上表现优越。 为了验证第 3.1 节讨论的超参数 设置策略以及 ALSS 模块中图 3 和图 4 第 A 部分所示的网络选择策略我们进行了如表 VI 所示的实验。具体而言
ALSS-YOLO’ 采用与 ALSS-YOLO 相反的 设置策略在低层特征层分配较大的 值以确保大部分特征通道通过图 3 和图 4 第 A 部分所示的网络结构而高层特征层则设置较小的 值。
ALSS-YOLO 则保持与 ALSS-YOLO 相同的 值但采用相反的网络结构选择策略即在低层特征层使用恒等连接在高层特征层使用卷积操作。
表 VI 数据显示ALSS-YOLO 的 mAP0.50 比 ALSS-YOLO’ 高 0.4%比 ALSS-YOLO 高 1.5%。这些结果验证了第 3.1 节的讨论并表明 ALSS-YOLO 的网络结构选择策略能够更有效地学习和表示输入数据的特征从而提升模型性能。 为研究在不同层集成 LCA 模块的影响我们基于 ALSS-YOLO 模型进行了一系列对比实验分析 LCA 模块分别置于第 8、18 和 22 层时的性能表现。这些配置分别标记为 ALSS-YOLO(LCAL8)、ALSS-YOLO(LCAL18) 和 ALSS-YOLO(LCAL22)。原始 ALSS-YOLO 模型也被纳入对比。实验结果总结于表 VII 中。结果显示ALSS-YOLO 在 mAP0.50 指标上取得了最高值并在其他指标上表现出强竞争力。而 ALSS-YOLO(LCAL22) 的 mAP0.50 最低表明该配置对整体模型性能产生负面影响。这些结果进一步验证了第 3.2 节中 LCA 模块位置选择的合理性。 为验证 ALSS-YOLO 模型的性能我们在BIRDSAI TIR UAV 数据集上与其他先进目标检测模型进行对比。这些模型包括 YOLOv3-tiny、YOLOX-nano、YOLOv5-n、YOLOv6-n、YOLOv8-ghost、YOLOv8-AM、YOLOv8-p2、MASK-RCNN-r18、FASR-RCNN-r18、RTMDet-tiny、VarifocalNet 和 CE-RetinaNet。为确保各模型性能对比的公平性所有模型均在相同训练环境下进行训练。ALSS-YOLO 与其他模型的对比结果见表 VIII性能最优模型的对比见图 16。 · ALSS-YOLO 模型在仅拥有 145.2 万参数的情况下取得了 mAP0.50 为 89.1% 的卓越表现展现了其在无人机应用中的高效性。 · 专为红外野生动物检测设计的 CE-RetinaNet 虽然获得了更高的 mAP0.5091.8%但其 5582.3 万的庞大参数量使其不适合在计算资源有限的无人机上部署。 · YOLOX-nano 拥有最小的参数量91.2 万但其检测精度比 ALSS-YOLO 低了 2.6%。 · 如图 16 所示的详细分析表明尽管 YOLOv8-p2 略胜一筹但参数量相当的 ALSS-YOLO-m 超越了 YOLOv8-p2 0.9%取得了 mAP0.50 为 90.3% 的优异成绩。
ALSS-YOLO 的较高召回率如图 16(b) 和 (c) 所示进一步突显了其鲁棒性。
这些发现清楚地表明ALSS-YOLO 能够有效应对 TIR 野生动物检测任务中的图像模糊和目标重叠等挑战。我们的模型在准确性与计算效率之间的平衡使其非常适合于基于无人机的监测任务。 为展示模型在不同类别上的适用性我们在典型场景条件下展示了各种物种的检测结果测试结果如图 17 所示。ALSS-YOLO-m 展现了优越的整体检测性能未出现漏检或误报。尽管 ALSS-YOLO 在某一场景中存在轻微漏检但其表现仍优于 YOLOv8-p2 等其他模型而后者在某些具有挑战性的场景中发生了显著漏检可能是由于特定场景条件或图像模糊详见表 VIII。这些分析证实 ALSS-YOLO 为无人机应用中的 TIR 野生动物检测提供了可靠的解决方案。 图 18 和 19 展示了模型在高度挑战性条件下的检测性能。在图 18 所示的高噪声干扰场景中ALSS-YOLO-m 模型实现了最高的召回率展现了卓越的检测性能。同时ALSS-YOLO-n 模型也取得了高度竞争力的结果表明其在挑战性条件下的有效性。图 19 所示的严重遮挡场景表明只有 ALSS-YOLO-m 模型能够检测到所有目标突出其在困难条件下的鲁棒性。 总结来说ALSS-YOLO在各个方面性能都有所提升。如果您有兴趣了解更多关于模型算法的使用方法等欢迎关注我们我们将继续为大家带来更多干货内容
