Positioning radiata pine branches requiring pruning by drone stereo vision
Organizations: Centre of Data Science and Artificial Intelligence & School of Engineering and Computer Science, Victoria University of Wellington, PO Box 600, Wellington, New Zealand · Computer Vision Research Lab & Department of Computer Science and Software Engineering, Canterbury University, PO Box 8041, Christchurch, New Zealand
Abstract
This paper presents a stereo-vision-based system mounted on a drone for detecting and localising radiata pine branches to support autonomous pruning. The proposed pipeline comprises two stages: branch segmentation and depth estimation. For segmentation, YOLOv8, YOLOv9, and Mask R-CNN variants are compared on a custom dataset of 71 stereo image pairs captured with a ZED Mini camera. For depth estimation, both a traditional method (SGBM with WLS filtering) and deep-learning-based methods (PSMNet, ACVNet, GWCNet, MobileStereoNet, RAFT-Stereo, and NeRF-Supervised Deep Stereo) are evaluated. A centroid-based triangulation algorithm with MAD outlier rejection is proposed to compute branch distance from the segmentation mask and disparity map. Qualitative evaluation at distances of 1-2 m indicates that the deep learning-based disparity maps produce more coherent depth estimates than SGBM, demonstrating the feasibility of low-cost stereo vision for automated branch positioning in forestry.
Figures & tables
| model | COCO datasets | Branches | ||
|---|---|---|---|---|
| mAP box50–95 | mAP mask50–95 | mAP box50–95 | mAP mask50–95 | |
| Mask R-CNN R50-C4 | 39.8 | 34.4 | 76.86 | 0.06 |
| Mask R-CNN R50-DC5 | 40.0 | 35.9 | 77.54 | 9.16 |
| Mask R-CNN R50-FPN | 41.0 | 37.2 | 79.19 | 6.75 |
| Mask R-CNN R101-C4 | 42.6 | 36.7 | 88.05 | 0.05 |
| Mask R-CNN R101-DC5 | 41.9 | 37.3 | 79.12 | 9.94 |
Stereo Vision
Stereo vision uses images from two or more viewpoints to estimate depth, recover 3D structure, and support spatial perception.
See all 43 papers