北京林业大学学报2025,Vol.47Issue(8):134-148,15.DOI:10.12171/j.1000-1522.20250151
基于树皮纹理的轻量化YOLOv11树种识别方法
Lightweight YOLOv11 tree species recognition method based on bark texture
摘要
Abstract
[Objective]To address the issue of existing tree species identification methods being difficult to deploy on mobile or edge devices with extremely limited hardware due to high computational complexity under varying lighting conditions,this study proposes a lightweight tree species identification method based on bark texture.[Method]This research improves YOLOv11 to construct the YOLOv11-SWER model.First,the lightweight feature extraction network StarNet was introduced as the backbone network,combining depthwise separable convolution and channel shuffle mechanisms to significantly reduce the model's parameter count and computational load during feature extraction.Second,a multi-branch feature fusion module,RepNCSPELAN4,was adopted,integrating group convolution and parameter-sharing strategies to balance global and local features,thereby enhancing multi-scale feature fusion efficiency.Then,a wavelet pooling(WaveletPool)layer was designed to reduce noise interference while preserving high-frequency texture details,improving the model's ability to capture subtle bark texture features.Finally,the detection head structure Detect_Efficient was optimized using a dual-branch group convolution architecture to improve computational efficiency.Additionally,based on a self-built dataset of 70 tree species with 6 681 bark images,ablation studies and comparative experiments were conducted to thoroughly evaluate the performance of improved model.[Result]The model achieved a detection precision of 98.1%,recall of 98.4%,mean average precision(mAP50)of 0.993,mean average precision across different IoU thresholds(mAP50-95)of 0.750,and an F1 score(harmonic mean of precision and recall)of 0.982.Compared with the YOLOv11 model,the parameter count and computational load were reduced by 46.92%and 51.5%,respectively,significantly lowering the model's spatial and computational complexity.It maintained stable identification performance under varying lighting conditions,demonstrating strong illumination robustness.[Conclusion]The proposed YOLOv11-SWER model,through lightweight design and multi-scale feature optimization,achieves high detection accuracy while reducing parameters by nearly half,striking a good balance between high accuracy and efficiency.This method holds promise for applications in intelligent forestry monitoring and urban forestry resource management.关键词
树皮纹理识别/YOLOv11/StarNet主干网络/小目标检测/轻量化Key words
bark texture recognition/YOLOv11/StarNet backbone network/small object detection/lightweight分类
信息技术与安全科学引用本文复制引用
张政银,向玮,刘子锋,王俊文,张咪,杨俊俐,黄泽园..基于树皮纹理的轻量化YOLOv11树种识别方法[J].北京林业大学学报,2025,47(8):134-148,15.基金项目
"十四五"国家重点研发计划(2022YFD2200502-04),北京高校大学生创新创业训练项目(202498117). (2022YFD2200502-04)