基于改进YOLO11n的林分小径木识别方法

孔硕 ,  秦昌浩 ,  张杨 ,  赵辉

森林工程 ›› 2026, Vol. 42 ›› Issue (03) : 570 -582.

PDF (13450KB)
森林工程 ›› 2026, Vol. 42 ›› Issue (03) : 570 -582. DOI: 10.7525/j.issn.1006-8023.2026.03.013
农林智能装备与技术

基于改进YOLO11n的林分小径木识别方法

作者信息 +

Improved YOLO11n-Based Recognition Method for Small-Diameter Trees in Forest Stands

Author information +
文章历史 +
PDF (13771K)

摘要

针对复杂林分中小径木树干细长、易遮挡且与背景纹理相似导致检测困难的问题,在自建小径木图像数据集基础上,提出基于YOLO11n构建轻量化改进模型——LDM-YOLO,LDM是由LSKA(large separable kernel attention)、DySample动态上采样与MBConv-Detect首字母缩写。方法上,在Backbone中引入双卷积跨阶段部分网络瓶颈‑大核可分离卷积注意力模块(cross‑stage partial network bottleneck with 2 convolutions-large separable kernel attention,C3k2-LSKA),增强细长树干及其上下文特征的表征能力;在Neck中采用DySample动态上采样,细化多尺度特征对齐;在检测头回归分支中使用移动倒置残差卷积‑检测结构(mobile inverted bottleneck convolution-detection head,MBConv-Detect),以降低参数量和计算量。试验结果表明,在自建小径木数据集上,提出的LDM-YOLO模型的mAP50和mAP50-95分别较YOLO11n提升3.1和2.3个百分点,GFLOPs从6.3 G降至5.1 G,在保持推理开销较低的前提下取得更高的检测精度。对比Faster R-CNN、YOLOv5n、YOLOv8n和YOLOv10n等代表算法,LDM-YOLO在相近或更低计算复杂度下取得了更优的综合检测性能。研究证明所提出模型在小径木检测任务中能够较好地兼顾精度与轻量化,为后续面向资源受限平台的林业智能装备应用提供了有益的参考。

Abstract

To address the detection difficulty of small-diameter tree trunks in complex forest stands—characterized by slender shapes, frequent occlusion, and high similarity to background textures—this study proposes a lightweight improved model, termed LDM-YOLO, built upon YOLO11n using a self-constructed small-diameter tree image dataset. The acronym LDM is derived from LSKA (large separable kernel attention), DySample dynamic upsampling, and MBConv-Detect. Specifically, a C3k2-LSKA module (cross-stage partial network bottleneck with two convolutions integrated with large separable kernel attention) is introduced into the Backbone to enhance the representation of slender trunks and their contextual features. In the Neck, DySample dynamic upsampling is adopted to refine multi-scale feature alignment. Moreover, an MBConv-Detect structure (mobile inverted bottleneck convolution–detection head) is applied to the regression branch of the detection head to reduce parameter count and computational cost. Experimental results on the self-built dataset show that LDM-YOLO improves mAP50 and mAP50–95 by 3.1 and 2.3 percentage points, respectively, compared with YOLO11n, while reducing GFLOPs from 6.3 G to 5.1 G, thereby achieving higher detection accuracy under low inference overhead. Compared with representative methods such as Faster R-CNN, YOLOv5n, YOLOv8n, and YOLOv10n, LDM-YOLO delivers superior overall detection performance at comparable or lower computational complexity. The results demonstrate that the proposed model better balances accuracy and lightweight design for small-diameter tree detection, providing a useful reference for subsequent forestry intelligent equipment applications on resource-constrained platforms.

Graphical abstract

关键词

小径木检测 / YOLO11 / 大核注意力 / 动态上采样 / 轻量化 / 特征增强 / MBConv / 林下复杂场景

Key words

Small-diameter tree detection / YOLO11 / large-kernel attention / dynamic upsampling / lightweight / feature enhancement / MBConv / complex understory scenes

引用本文

引用格式 ▾
孔硕,秦昌浩,张杨,赵辉. 基于改进YOLO11n的林分小径木识别方法[J]. 森林工程, 2026, 42(03): 570-582 DOI:10.7525/j.issn.1006-8023.2026.03.013

登录浏览全文

4963

注册一个新账户 忘记密码

0 引言

在现代可持续林业管理中,对林分结构的优化至关重要。林分中普遍存在胸径较小、树干细长的幼树及下层细小树干(本研究统称为“小径木”),这类个体生长势相对较弱但数量较多,通常被作为选择性砍伐与抚育作业的主要对象,用以调节林木密度、改善林分整体结构,进而促进优势目标木的健壮生长[1]。然而,传统的人工巡查与样地调查方法效率低下、成本高昂,且难以满足大面积、高精度的监测需求,已成为制约现代林业向自动化、智能化发展的关键瓶颈。近年来,计算机视觉与深度学习技术的飞速发展为解决这一难题提供了全新的技术路径[2]。

以卷积神经网络(convolutional neural network,CNN)为核心的深度学习模型,能够从海量图像数据中自动学习和提取高维、鲁棒性的特征,摆脱了传统图像处理方法对人工设计特征的依赖。单阶段目标检测算法YOLO系列[3]凭借端到端高速检测和较高准确率,已成为林业树干检测的主流选择。例如,Badgujar等[4]采用YOLOv5对天然牧场环境中的东红杉树干进行检测,经过迁移学习训练的模型在测试中达到约87.8%的精度、84.3%的召回率和88.9%的平均精度,能够识别不同粗细的树干并有效区分树干与树枝。在密林场景中,Da Silva等[5]比较了多种算法,结果表明YOLOv7及其变体YOLOR在大规模林地树干数据集上取得了最高约90%的F1分数,而且在图形处理器(graphics processing unit,GPU)上仍可实现毫秒级的实时检测。针对更复杂的应用需求,Liu等[6]在YOLOv7中融入注意力机制并改进损失函数,用于油茶林收获机器人视觉中的树干识别,使模型平均精度提升至89.2%、召回率提高到94%,检测性能显著优于原始YOLOv7模型。Zhang等[7]在YOLOv8基础上引入金字塔特征增强网络、Transformer模块和坐标注意力机制,提出了PCC-YOLO金字塔增强网络、上下文Transformer模块、坐标-通道融合注意力机制YOLO(a YOLOv8-based model integrating PENet,CoT-Net,and Coord-SE attention,PCC-YOLO)树干检测模型,在果园树干数据集上实现了82.6%的mAP和约143帧/s的检测速度,表现出卓越的精度与效率平衡。Zhang等[8]提出了面向林地场景的轻量级YOLOv11-TrunkLight模型,引入新型高效骨干网络和可变形注意力模块,有效提升了树干目标的检测性能。杨灿等[9]提出空间感知与多尺度特征增强自适应网络YOLO(spatial‑aware and FastScaleNet YOLO,SSFYOLO),通过空间感知模块(spatial-aware module,SAM)、多尺度特征增强网络FastScaleNet与加权IoU机制,显著增强了模型在复杂林地小目标检测中的精度与稳定性。胡峻峰等[10]针对森林移动机器人算力受限与光照变化大的问题,提出基于YOLOv7-tiny的轻量级检测方法,引入部分通道卷积高效层聚合网络(partial efficient layer aggregation network,P-ELAN)与α‑完整交并比(alpha-complete Intersection over union,alpha-CioU),提升推理速度同时保持检测精度,适配边缘设备部署。在农业无人机低空作业场景下,徐相华等[11]提出双坐标注意力-单阶段目标检测模型(dual coordinate attention you only look once,DCA-YOLO),引入动态通道注意力与轻量化主干网络,实现复杂背景下对病虫害树木的高效精准识别。此外,针对与树干结构高度关联的树木年轮信息提取任务,王品博等[12]提出基于YOLOv8的DCW-YOLOv8)可扩张残差注意力模块-轻量级通用上采样算子-动态非单调聚焦机制边界框损失函数的YOLOv8模型(dilation-wise residual-content-aware reassembly of features-weighted interpolation of sequential evidence for intersection over union,DCW-YOLOv8),实现了年轮实例分割与宽度测量,为林木生长特征自动化识别提供了新思路。

尽管现有研究成果已在树干检测领域取得显著进展,但在处理林分复杂场景,尤其是小径木目标时仍存在挑战。一方面,小径木树干直径较小、形态差异明显,易受枝叶遮挡、光照变化和背景纹理干扰影响,深度学习模型对细小目标的识别鲁棒性仍显不足;另一方面,高精度检测模型多依赖高算力硬件,对于森林巡检机器人、便携式巡护设备或移动端部署而言,其实时性和能耗需求难以同时满足。基于上述问题,为同时提升小径木检测精度与模型可部署性,本研究在YOLO11n的基础上构建改进模型LDM-YOLO:在Backbone中引入双卷积跨阶段部分网络瓶颈-大核可分离卷积注意力模块(cross‑Stage partial network bottleneck with 2 convolutions-large separable kernel attention,C3k2-LSKA)以增强细长树干及其上下文信息建模能力;在Neck中采用DySample动态上采样缓解特征对齐不准问题;并在检测头中设计轻量化移动倒置残差卷积‑检测结构(mobile inverted bottleneck convolution-detection head,MBConv-Detect),以在降低参数量和计算量的同时,提升对小径木树干边界和内部纹理的刻画能力,LDM是由LSKA(large separable kernel attention)、DySample动态上采样与MBConv-Detect首字母拼写。

1 LDM-YOLO模型构建

1.1 YOLO11网络模型

YOLO11是Ultralytics团队于2024年发布的新一代单阶段目标检测模型,提供纳米级模型(nano,n)、小型模型(small,s)、中型模型(medium,m)、大型模型(large,l)、超大型模型(extra‑large,x)多种规模以兼顾检测精度与计算开销,整体网络仍由Backbone、Neck和Head构成。Backbone以改进的C3k2模块替代C2f,结合跨阶段部分连接思想与多分支卷积/特征拼接,在控制参数量的同时增强特征重用与梯度传播;末端引入简化空间金字塔池化快速模块(simplified spatial pyramid pooling fast,SPPF)与C2PSA模块以获取多尺度上下文并强化空间注意,从而提升复杂背景和小目标的表征能力。Neck采用改进的路径聚合网络‑特征金字塔网络结构(path aggregation network-Feature pyramid network,PAN-FPN),通过上采样、下采样及特征拼接实现多尺度特征融合,并在各尺度分支继续使用C3k2以优化跨层信息传递。Head采用无锚设计,在多尺度特征图上直接进行回归与分类预测,由C3k2与标准卷积(conv‑batchnorm‑SiLU,CBS)组合,对各尺度输出边界框参数、置信度和类别概率,实现高效端到端检测。

1.2 LDM-YOLO模型

在基线网络YOLO11n的基础上,本研究结合小径木“细长、易遮挡、与背景相似”的特点,从主干网络、特征融合和检测头3个层面对模型进行了有针对性地改进:在Backbone中将部分C3k2单元替换为C3k2-LSKA模块,以增强对细长树干及其上下文信息的建模能力;在Neck中用DySample动态上采样替代原有固定插值,上采样过程中更好地对齐多尺度特征;在解耦检测头中设计轻量化MBConv-Detect结构,在控制参数量和计算量的同时提升对小径木边界和内部纹理的刻画能力。整体网络结构见图1,改进后的LDM-YOLO在保持推理速度基本不变的前提下,使模型在遮挡和密集林下场景中的小径木检测精度得到进一步提升。

1.2.1 LSKA模块

小径木目标在图像中呈现出极端的纵向细长形态,且常被林下灌木或枝叶沿树干方向不规则遮挡,导致目标特征在空间上呈现非连续的断裂状态。传统的小卷积核由于感受野受限,难以捕获树干的长距离上下文依赖,极易将断裂的树干误判为背景噪声或多个独立目标。为此,本研究在YOLO11n主干网络中引入大核可分离卷积注意力(large separable kernel attention,LSKA)模块[13],对部分C3k2基本单元进行替换,构建C3k2-LSKA结构。LSKA通过大核卷积模拟人类视觉对细长目标的整体感知机制,能够在更大的空间范围内聚合上下文信息,从而有效跨越局部遮挡区域,将被打断的树干特征重新关联,增强模型对细长目标形态的完整性感知。其整体结构见图2。LSKA本质上是一种“大核+可分离深度卷积+空间注意力”的结构,通过将二维k×k大核分解为级联的1×k与k×1一维核,在近似保持大感受野的同时有效降低参数量和 GFLOPs。

设输入特征为 F,LSKA首先采用一对水平方向和垂直方向的深度卷积提取局部上下文,再进一步结合空洞深度卷积建模长程依赖关系。以第c个通道为例,其计算过程可表示公式为

Z˜c=W(2d-1)×1c*W1×(2d-1)c*Fc。
Zc=Wk/d×1c*W1×k/dc*Z˜c。
Ac=W1×1c*Zc。
F^c=Ac⊙Fc。

式中:c为通道索引;Fc表示第c个通道特征;*表示卷积运算;⊙表示Hadamard逐元素乘积;Wa×bc为第c个通道对应的卷积核,尺寸为a×b;Z˜c为经水平和垂直深度卷积后的中间特征;Zc为进一步结合空洞深度卷积后的特征;Ac为生成的注意力权重;F^c为加权后的输出特征;k为最大感受野尺寸参数;d为空洞率。

上述过程等价于使用一个大核深度卷积生成空间注意力权重Ac,再与特征Fc相乘实现自适应特征重标定。由于二维大核被分解为4个一维核,LSKA的参数量和计算复杂度随k近似线性增长,相比直接使用k×k深度卷积或传统大核注意力(large kernel attention,LKA)[14]的二维大核,可显著降低GFLOPs和显存占用。

在具体实现上,对于浅层特征,将C3k2中的残差块替换为LSKA注意力单元,利用大核可分离卷积,该模块能够在较高分辨率上有效捕获树干的纵向延展与弯曲形态;对于深层特征,考虑到通道数显著增加、计算成本更为敏感,则保留原有的C3k2结构,仅在浅层引入LSKA,以在检测精度与推理效率之间取得平衡。整体而言,C3k2-LSKA既保持了C3k2良好的梯度传递和特征复用能力,又通过大核可分离注意力机制增强了对细长小径木目标的形状建模与全局上下文感知能力。

1.2.2 动态上采样模块DySample

在小径木检测场景中,树干往往细长、弯曲且尺度差异明显,简单的最近邻或双线性上采样容易造成细窄树干被模糊、断裂,难以兼顾粗壮树干与细小枝干的结构细节。在特征金字塔的融合过程中,小径木目标由于像素占比极低,对上采样算法的插值误差极为敏感。传统的最近邻或双线性插值采用固定的采样核,容易造成细微的树干纹理被平滑模糊,甚至导致细小目标在深层特征映射回高分辨率时发生结构丢失。为此,引入动态上采样模块DySample[15]替代固定插值。DySample的核心优势在于其内容感知机制,能够根据小径木树干的纹理分布自适应调整采样点位置,相当于沿着树干的边缘与骨架进行精准重采样。这种机制能有效避免细小目标边缘的模糊化,最大限度地保留小径木在复杂背景下的高频结构信息。DySample的整体结构见图3。

对于输入特征图 x,DySample采用双分支线性变换分别建模“动态缩放”和“静态分布”两类信息。2条支路的线性映射可表示为

L1(X)=W1X+b1。
L2(X)=W2X+b2。

式中:L1(X)更关注局部高频、变化剧烈的区域;L2(X)则描述整体较平滑的静态特征分布;b1与b2为2条线性映射的偏置项,用于为分支输出提供可学习的常数平移,使线性变换的输出不必经过原点,从而调节特征响应的基线水平并提高映射的表达能力与训练稳定性为控制动态偏移量的幅度,对动态分支进行抑制后与静态分支融合,得到中间特征Pi,其计算公式为

F1=0.5α⊙L1(X)。
Pi=F1+L2(X)。

这一设计一方面利用L1(X)强调边缘和纹理等细节,另一方面通过L2(X)保持整体结构的稳定性,两者在Pi中实现互补,有利于在后续上采样过程中避免偏移量过大导致采样位置发散。随后,DySample依托像素重排(PixelShuffle)完成亚像素级上采样,将通道维度的冗余映射为更高的空间分辨率。由中间特征Pi得到的静态偏移可表示公式为

O=PixelShuffle(Pi,s)。

式中: O 为由PixelShuffle将中间特征Pi重排到高分辨率空间后得到的偏移。

为获得最终采样位置集合,将原始网格 G 与偏移 O 相加得到采样集合 S,其计算公式为

S=O+G。

在LDM-YOLO中,将预测头P4→P3、P5→P4这2条上采样路径的固定上采样算子替换为DySample,以在保持轻量级结构的前提下,使网络能够根据小径木目标的空间分布自动调整采样位置。

1.2.3 基于MBConv的轻量解耦检测头

由图4可知,原始YOLO11n检测头采用解耦结构:上分支由2层3×3卷积和1层1×1卷积构成,用于回归边界框;下分支由交替堆叠的3×3深度可分离卷积(DWConv)和1×1卷积构成,用于类别预测。该设计能够在一定程度上实现回归与分类的任务解耦,但回归分支仍大量使用普通卷积,参数量和计算量相对较大,对细长、弯曲的小径木边界的建模能力仍有进一步提升空间。

为在保持检测头轻量化的同时增强回归分支对小径木目标的表征能力,本研究引入EfficientNet系列中的MBConv结构[16],对YOLO11n的检测头进行改造。MBConv采用“扩展—深度卷积—压缩”的倒残差设计:首先通过1×1卷积在通道维度上对特征进行扩展,随后使用3×3深度可分离卷积提取空间细节,并在中间嵌入SE通道注意力模块以实现特征重标定,最后通过1×1卷积完成通道压缩,并与输入进行残差连接,见图5。该结构一方面利用深度可分离卷积显著降低参数量和GFLOPs,另一方面借助SE与残差连接提升有效感受野和梯度传递能力,兼顾了轻量化与表达能力。

基于上述思想,本研究构建了MBConv-Detect检测头。在林下复杂环境中,小径木树皮纹理往往与背景灌木高度相似,导致特征辨识度低。MBConv结构中的SE注意力机制能够对特征通道进行自适应重标定,抑制背景噪声通道的响应,增强包含树干纹理特征的通道权重,从而在特征层面实现“去伪存真”。同时,倒残差结构在保持低参数量的同时维持了深层特征的非线性表达能力。在回归分支中引入MBConv,不仅大幅降低了检测头的计算负载,更通过特征重标定机制提升了对微弱纹理目标的边界回归精度,实现了轻量化与高精度的双重优化。

2 材料与设置

2.1 小径木图像数据集构建

本研究于2025年9—11月在东北林业大学城市林业示范基地(45°43′25.50″ N,126°37′51.05″ E)开展多批次实地采集,使用配备高分辨率摄像头的智能手机(iPhone 15)进行地面近景拍摄。采集过程中重点覆盖生长弯曲、胸径较小和密集成丛等典型小径木个体,并兼顾不同光照条件与背景复杂度,以增强数据集对实际作业场景的代表性。对原始图像进行人工筛选,剔除模糊、严重遮挡及目标轮廓不清晰等低质量样本,最终保留1 600张高质量小径木图像作为有效原始样本。为了明确任务边界并保证试验的可复现性,本研究结合林业幼林抚育需求与计算机视觉任务特性,制定了小径木视觉检测标准:定义检测对象为地径或胸径在1~4 cm的林下幼树及灌木状细小林木;视觉特征层面,仅对图像中纹理特征可辨且未被枝叶完全遮挡的树干区域进行标注。对于远处背景中像素宽度过小或因运动模糊导致轮廓严重缺失的极微小目标,视为背景噪声不予标注,以降低标签的不确定性对模型训练的干扰。使用LabelImg对全部图像逐张标注,将小径木树干作为唯一检测目标,以矩形边界框进行标记,并按YOLO格式生成标注文件,类别统一命名为“小径木(sdtree)”。

为提升模型在不同成像条件下的鲁棒性与泛化能力,对原始样本实施数据增强,包括水平翻转以模拟视角变化,图像锐化以强化树干边缘与纹理细节,以及适度调整饱和度和色调以模拟不同光照与环境色温。经增强后,数据集规模由1 600张图像扩展至3 044张图像,并按8∶1∶1的比例随机划分为训练集、验证集和测试集,用于后续模型训练与性能评估。部分小径木图像样本见图6。

2.2 试验环境与参数设置

本研究的试验硬件配置为Intel Core i9-13900HX处理器、32 GB内存、NVIDIA GeForce RTX 5070 Ti独立显卡,操作系统为Windows 11。软件环境采用Python 3.11.13、PyTorch 2.8与CUDA 12.8,目标检测算法基于Ultralytics YOLO框架实现。训练时,采用官方提供的YOLO11n预训练权重作为初始化参数,以加快收敛并提升特征表达能力。输入图像尺寸为640×640像素,batchsize设为16,训练轮数为150,初始学习率为0.01,优化器采用随机梯度下降(stochastic gradient descent,SGD),其余超参数设置遵循Ultralytics YOLO的默认配置。

2.3 评估指标

本研究采用精度(precision,P)、召回率(recall,R)、平均精度(average precision,AP,式中记为AP)和平均精度均值(mean average precision,mAP,式中记为mAP)评价各模型在小径木检测任务中的综合性能,并同时给出mAP50与mAP50-95。精度表示被判为正类的样本中真正例的比例,召回率表示所有真正例中被正确检出的比例,二者均可由混淆矩阵中的True Positive(式中记为TP)、False Positive(式中记为FP)和False Negative(式中记为FN)计算得到。

在目标检测中,AP表征某一类别在不同召回率水平下的整体检测性能,定义为精度-召回率(P-R)曲线在召回率区间上的面积;mAP为各类别AP的平均值,是衡量模型总体检测效果的核心指标。本研究中,mAP50对应IoU阈值为0.5时的mAP,mAP50-95则在IoU为0.5-0.95(步长0.05)下分别计算AP后取平均,用以综合反映不同重叠要求下的检测性能。为刻画模型复杂度与推理开销,本研究同时报告参数量(params)、浮点运算量(GFLOPs)及模型权重大小(model size),分别用于衡量可学习参数规模、单次前向推理的近似计算量以及模型的存储占用。

本研究中上述评价指标的计算公式为

P=TPTP+FP。
R=TPTP+FN。
AP=∑k=1KPkΔRk。
mAP=1Ncls∑i=1NclsAPi。

式中:Pk为第k个采样点的精度;ΔRk为相邻2点召回率增量;Ncls为类别数;i为类别索引;APi为第i类目标的平均精度。

3 结果与分析

3.1 模型训练收敛性分析

为了验证改进模型在训练过程中的收敛性能与泛化能力,本研究记录了YOLO11n基线模型与LDM-YOLO改进模型的训练损失与验证损失变化曲线,见图7。由图7(a)的训练损失曲线可以看出,LDM-YOLO在训练初期的下降速度快于基线模型,且在整个训练周期内始终保持更低的损失值,表明引入的C3k2-LSKA模块有效加速了网络对小径木特征的提取与拟合。由图7(b)的验证损失曲线可见,2种模型均在训练至80轮左右进入相对平稳的收敛阶段,且未出现验证损失持续上升的过拟合现象。值得注意的是,在第140轮处,由于训练策略关闭了Mosaic数据增强,2条曲线均出现了预期的短暂波动并迅速回落,进一步证明了模型对真实数据分布的适应能力。相比基线模型,LDM-YOLO的验证损失更低,说明改进后的模型在非结构化林业场景中具有更强的鲁棒性。

3.2 消融试验

为系统评估3种改进模块对YOLO11n在小径木检测任务上的作用,现对表1中的消融试验结果进行分析。表1中第1组为未引入任何改进模块的YOLO11n基线模型,第2至第6组在此基础上按列打“√”表示启用了相应模块。与基线相比,单独加入C3k2-LSKA(第2组)和DySample(第3组)均能显著提升检测精度:P分别提高至75.9%和77.9%,mAP50提升到77.6%和77.1%,mAP50-95也增加了1.8和2.7个百分点,说明大卷积核注意力与动态上采样均有利于在复杂林分背景下强化小径木目标的特征表达与空间细节;引入MBConv-Detect(第4组)时,mAP基本保持不变,而GFLOPs由6.3 G降至5.1 G、参数量由2.58 M降至2.26 M、权重由5.2 MB缩减到4.6 MB,表明该检测头在保证精度的前提下有效实现了模型轻量化。

组合试验中,C3k2-LSKA与DySample同时启用(第5组)使mAP50达到78.0%的最高值,验证了二者在全局建模与上采样细节方面的互补性;在此基础上进一步加入MBConv-Detect(第6组)后,P、R以及各项mAP指标仍整体优于基线,同时保持GFLOPs的5.1 G和2.27 M参数量,实现了检测性能与模型复杂度之间较为理想的折中,表明三种模块在Backbone、Neck与Head这三个层面形成协同作用,使改进后的YOLO11n在小径木检测任务上兼顾了精度提升与轻量化需求。

3.3 对比试验

为进一步评估所提出LDM-YOLO模型在小径木检测任务中的综合性能,本研究选取Faster R-CNN[17]、YOLOv5n、YOLOv8n、YOLOv10n以及未改动的YOLO11n作为对比模型,在相同数据集划分、输入分辨率和训练策略下进行试验,结果见表2。由表2可以看出,LDM-YOLO在所有模型中取得了最高的精度和召回率,mAP50和mAP50-95也分别达到77.4%和39.2%,相较于YOLO11n分别提升3.1和2.3个百分点。

与两阶段检测器Faster R-CNN相比,在显著降低计算复杂度的前提下优势更为明显:Faster R-CNN的GFLOPs与参数量高达133.92 G和41.3 M,而LDM-YOLO仅为5.1 G和2.27 M,却在mAP50与mAP50-95上分别高出0.8和4.3个百分点,说明基于C3k2-LSKA、DySample与MBConv-head的改进结构能够在复杂林分场景中更有效地表征小径木目标。与轻量化单阶段检测器相比,LDM-YOLO在保持类似参数规模的同时也表现出更为优越的精度表现。相较于YOLOv8n,本模型在mAP50和mAP50-95上分别提升了4.4和4.6个百分点。这主要归因于YOLOv8n采用的标准C2f模块和最近邻上采样在处理极细树干时,容易因感受野受限和插值锯齿效应导致纹理细节丢失,而LDM-YOLO的LSKA大核注意力和DySample动态上采样机制则能更精准地保留细长目标的边缘结构。此外,针对YOLOv10n的试验结果显示,尽管其GFLOPs高达8.2 G,但mAP50却比LDM-YOLO低9.2个百分点。深入分析可知,YOLOv10n虽然引入了PSA大核卷积与无非极大值抑制(non-maximum suppression,NMS)策略,但其通用特征提取结构在应对纹理匮乏的小径木时出现了计算资源错配现象,大量算力被消耗在背景特征的冗余计算上,未能像LDM-YOLO一样实现对细长前景目标的算力聚焦。

针对资源受限平台的部署需求,本研究在统一640×640输入分辨率下进行了推理速度测试。LDM-YOLO的单张推理耗时为4.8 ms,对应的检测速度约为208帧/s。与基线模型YOLO11n相比,推理时间减少了约1.5 ms;与Faster R-CNN相比,推理速度有显著优势。

综合来看,LDM-YOLO在保证模型轻量化的前提下,实现了在小径木数据集上的最高精度和较高的定位准确度,验证了所提出的3种结构改进在实际小径木检测场景中的有效性与实用价值。

为进一步探究改进模型在具体类别上的识别优势,本研究绘制了YOLO11n与LDM-YOLO在测试集上的归一化混淆矩阵,见图8。混淆矩阵的对角线数值代表模型对该类别的识别准确率(recall),数值越高表明漏检越少。由图8可知,基线模型YOLO11n对小径木(sdtree)类别的识别率为0.72,这意味着有28%的真实小径木目标被误判为背景。而改进后的LDM-YOLO模型将小径木的识别率提升至0.77,漏检率降低了5个百分点。这一提升主要得益于DySample动态上采样与MBConv-Detect检测头的引入,有效增强了模型对细小、被遮挡树干目标的特征对齐与边界感知能力,使其在纹理复杂的林下背景中能更精准地分离出前景目标。

3.4 结果可视化分析

为了直观揭示模型在复杂林分环境下对小径木目标的关注区域,本研究采用了一种由检测先验驱动的特征可视化方法。该方法首先在推理阶段提取网络关键层的深层特征响应,并将其映射为空间激活强度;然后利用非线性增强技术突出显著特征区域的对比度;最后结合模型预测的边界框信息引入空间位置约束,对原始激活图进行加权融合。该处理策略能够有效滤除非目标区域的背景噪声干扰,从而更具针对性地呈现模型对小径木目标主体特征的聚焦效果。

对3幅典型场景进行热力图可视化,见图9。其中,第1行为原始图像,第2行为YOLO11n的热力图结果,第3行为LDM-YOLO的热力图结果,颜色越接近红色表示响应越强。从第1列、第2列可以看到,2种模型都能在目标树干区域形成纵向高响应带,但LDM-YOLO的高响应更加连续,热点集中在树干主体及其分叉处,背景灌木和地面叶片区域基本被压制为冷色;相比之下,YOLO11n的响应分布略显发散,部分能见度较低的小径木只呈现出零散的弱响应。在第3列(复杂背景、遮挡较重)中差异更为明显:YOLO11n在目标位置几乎没有明显高响应,整体呈冷色,容易造成漏检;而LDM-YOLO仍能在细长树干上形成清晰的纵向暖色条带,说明改进后的网络在林下杂乱环境中,对细小、背景相近的小径木具有更强的聚焦能力和特征分离能力。

为了对比不同模型在真实林下环境中的检测表现,本研究选取了“简单场景、遮挡场景和密集场景”3幅代表性图像进行可视化,见图10。整体来看,YOLOv5n、YOLOv8n、YOLOv10n、YOLO11n和LDM-YOLO均能在3种场景下检测出主要小径木目标,但置信度和框的位置精度存在明显差异:YOLOv5n、YOLOv8n在遮挡和密集场景中部分目标的置信度偏低,YOLOv10n在密集场景中出现了小径木漏检现象。相比之下,YOLO11n与LDM-YOLO的检测框更贴合树干轮廓,其中LDM-YOLO在3种场景下均表现出更高且更稳定的置信度,说明改进模型在复杂背景、遮挡和多目标条件下具有更好的检测可靠性。

4 结论

本研究针对复杂林分环境中小径木“细长、易遮挡、与背景相似”等特征带来的检测难题,在YOLO11n的基础上构建了轻量化改进模型LDM-YOLO。通过在Backbone中引入C3k2-LSKA模块,在Neck中采用DySample动态上采样,以及在检测头回归分支中使用MBConv-Detect结构,模型在多尺度特征表达、细长树干边界分析和参数压缩等方面得到协同优化。试验及对比结果表明,LDM-YOLO在mAP50和mAP50-95上均较基线模型与典型对比算法取得了更优性能,同时降低了GFLOPs和参数量。这证明了本研究提出的改进策略能够更有效地聚焦细长、被遮挡及密集分布的小径木目标,较好地兼顾了检测精度与模型轻量化需求。

尽管LDM-YOLO在本试验中实现了精度与速度的良好平衡,但本研究仍存在一定局限性。自建数据集(3 044张)规模偏小且场景采集范围相对集中,主要反映了特定林分与季节的特征,限制了模型在异构植被或极端气候下的泛化能力。面向广泛场景部署时,还需采用迁移学习或微调策略以增强模型的环境适应性。热力图分析显示,在极度遮挡或树皮纹理与背景高度相似的极端场景下,仅依赖RGB特征仍难以完全避免漏检。这表明可见光感知在复杂林下环境中存在物理瓶颈。

针对上述不足,未来工作可进一步扩展数据集场景与季节范围,引入多类别林木与障碍物检测;同时考虑引入深度信息或多光谱数据进行多模态融合,利用三维空间与光谱差异进一步提升检测稳定性。综上所述,LDM-YOLO模型具有一定的实用价值,能够为后续在林业移动机器人、便携式巡护设备等资源受限平台上开展小径木智能识别提供可靠的算法基础与参考。

参考文献

[1]

白宇,杨华,温静,等.基于邻近木的林分结构多样性研究[J].北京林业大学学报,2020,42(6):52-58.

[2]

BAI Y, YANG H, WEN J,et al.Study on forest structure diversity based on the neighbourhood trees[J].Journal of Beijing Forestry University,2020,42(6):52-58.

[3]

ZHAO Z Q, ZHENG P, XU S,et al.Object detection with deep learning:A review[J].IEEE Transactions on Neural Networks and Learning Systems,2019,30(11):3212-3232.

[4]

REDMON J, DIVVALA S, GIRSHICK R,et al.You only look once:Unified,real-time object detection[C]//2016 IEEE Conference on Computer Vision and Pattern Recognition.June 27-30,2016,Las Vegas,NV,USA,IEEE,2016:779-788.

[5]

BADGUJAR C, FLIPPO D, GUNTURU S,et al.Tree trunk detection of eastern red cedar in rangeland environment with deep learning technique[J].Croatian Journal of Forest Engineering,2023,44(2):357-368.

[6]

SILVA D Q DA, DOS SANTOS F N, FILIPE V,et al.Edge AI-based tree trunk detection for forestry monitoring robotics[J].Robotics,2022,11(6):136.

[7]

LIU Y, WANG H R, LIU Y H,et al.A trunk detection method for Camellia oleifera fruit harvesting robot based on improved YOLOv7[J].Forests,2023,14(7):1453.

[8]

ZHANG S, SUN Y, QIAO Y,et al.PCC-YOLO:A trunk detection algorithm for orchard robots based on improved YOLOv8[J].Agriculture,2025,15(16):1786.

[9]

ZHANG S Q, ZHENG Y B, BI R G,et al.Trunk detection in complex forest environments using a lightweight YOLOv11-TrunkLight algorithm[J].Sensors,2025,25(19):6170.

[10]

杨灿,范习健,张九于.SSFYOLO:一种面向复杂森林场景的树干检测算法[J].北京林业大学学报,2025,47(2):132-142.

[11]

YANG C, FAN X J, ZHANG J Y.SSFYOLO:A trunk detection algorithm for complex forest scenarios[J].Journal of Beijing Forestry University,2025,47(2):132-142.

[12]

胡峻峰,朱昊,黄晓文,等.基于深度学习的森林移动机器人树干检测[J].森林工程,2024,40(4):109-114.

[13]

HU J F, ZHU H, HUANG X W,et al.Trunk detection for forest mobile robots based on deep learning[J].Forest Engineering,2024,40(4):109-114.

[14]

徐相华,周德靖,余超然,等.基于DCA-YOLO的受病虫害侵染树木农业无人机低空检测模型[J].农业机械学报,2025,56(10):479-491.

[15]

XU X H, ZHOU D J, YU C R,et al.Improved model for low altitude detection of trees infected by pests and diseases using agricultural drones based on DCA-YOLO[J].Transactions of the Chinese Society for Agricultural Machinery,2025,56(10):479-491.

[16]

王品博,王雨晨,王双永,等.基于改进YOLOv8的树木年轮实例分割及信息检测方法[J].木材科学与技术,2025,39(3):7-18.

[17]

WANG P B, WANG Y C, WANG S Y,et al.Study on tree ring instance segmentation and information detection based on improved YOLOv8[J].Chinese Journal of Wood Science and Technology,2025,39(3):7-18.

[18]

LAU K W, PO L M, REHMAN Y A U.Large separable kernel attention:rethinking the large kernel attention design in CNN[J].Expert Systems With Applications,2024,236:121352.

[19]

AZAD R, NIGGEMEIER L, HUTTEMANN M,et al.Beyond self-attention:Deformable large kernel attention for medical image segmentation[C]//2024 IEEE/CVF Winter Conference on Applications of Computer Vision.2024:1287-1297.

[20]

LIU W Z, LU H, FU H T,et al.Learning to upsample by learning to sample[C]//2023 IEEE/CVF International Conference on Computer Vision.2023:6027-6037.

[21]

TAN M X, LE Q V.EfficientNetV2:Smaller models and faster training[C]//2021 International Conference on Machine Learning.2021:10096-10106.

[22]

REN S Q, HE K M, GIRSHICK R,et al.Faster R-CNN:towards real-time object detection with region proposal networks[C]//IEEE Transactions on Pattern Analysis and Machine Intelligence.IEEE,2015,28:1137-1149.

基金资助

黑龙江省应用技术研究与开发计划项目(GA21A403)

2025年高教强省专项-产教融合研究生联合培养基地建设资助项(41502123)

AI Summary AI Mindmap
PDF (13450KB)

478

访问

0

被引

详细

导航
相关文章

AI思维导图

/

〈 〉