YOLO and SGBM Integration for Autonomous Tree Branch Detection and Depth Estimation in Radiata Pine Pruning Applications
Yida Lin, Bing Xue, Mengjie Zhang, Sam Schofield, Richard Green
TL;DR
This work addresses safe, cost-effective autonomous pruning of radiata pine by integrating YOLO-based branch detection with SGBM stereo depth estimation, eliminating the need for LiDAR. It demonstrates superior branch segmentation performance over Mask R-CNN, achieving high mAP scores, and validates sub-second processing with reliable 3D localization within a 2 m range. The study presents a complete pipeline from data collection to 3D localization, including data augmentation, depth refinement via WLS, and robust spatial registration. The results show the approach is feasible for real-time autonomous pruning on consumer-grade hardware, offering safety enhancements and economic viability for forestry operations. Overall, the framework advances autonomous forestry by delivering accurate, low-cost perception and depth estimation suitable for branch-aware navigation and manipulation.
Abstract
Manual pruning of radiata pine trees poses significant safety risks due to extreme working heights and challenging terrain. This paper presents a computer vision framework that integrates YOLO object detection with Semi-Global Block Matching (SGBM) stereo vision for autonomous drone-based pruning operations. Our system achieves precise branch detection and depth estimation using only stereo camera input, eliminating the need for expensive LiDAR sensors. Experimental evaluation demonstrates YOLO's superior performance over Mask R-CNN, achieving 82.0% mAPmask50-95 for branch segmentation. The integrated system accurately localizes branches within a 2 m operational range, with processing times under one second per frame. These results establish the feasibility of cost-effective autonomous pruning systems that enhance worker safety and operational efficiency in commercial forestry.
