用YOLO11和深度模型精准定位果园幼果3D姿态,助力机器人疏果
YOLO11 and Vision Transformers based 3D Pose Estimation of Immature Green Fruits in Commercial Apple Orchards for Robotic Thinning
- 结合YOLO11与ViT深度模型进行幼果3D姿态估计
- YOLO11n在精度和速度上均优于YOLOv8系列,推理仅2.7毫秒
- Depth Anything V2在果长估计中误差最低,适合高精度机器人应用
本研究提出一种鲁棒的商业苹果园中未成熟绿苹果(果蕾)3D姿态估计方法,结合YOLO11目标检测与姿态估计算法及视觉变换器(ViT)进行深度估计(DPT与Depth Anything V2)。在相同超参数设置下,对YOLO11n、YOLO11s、YOLO11m、YOLO11l、YOLO11x与YOLOv8n、YOLOv8s、YOLOv8m、YOLOv8l、YOLOv8x进行性能对比。结果显示,YOLO11n在框精度和姿态精度上分别达到0.91和0.915,领先所有配置;而YOLOv8n在框召回率(0.905)和姿态召回率(0.925)上最优。在mAP@50方面,YOLO11s框识别得分最高(0.94),YOLOv8n姿态识别得分最高(0.96)。图像处理速度上,YOLO11n仅需2.7毫秒,远快于最快的YOLOv8n(7.8毫秒)。进一步集成ViT进行深度估计发现,Depth Anything V2在3D姿态长度验证中表现最佳,根均方误差(RMSE)为1.52,平均绝对误差(MAE)为1.28,展现出优异的幼果长度估计精度。该方法为机器人疏果提供了高效可行的3D姿态估计解决方案。
原文摘要 · Abstract (English)
In this study, a robust method for 3D pose estimation of immature green apples (fruitlets) in commercial orchards was developed, utilizing the YOLO11(or YOLOv11) object detection and pose estimation algorithm alongside Vision Transformers (ViT) for depth estimation (Dense Prediction Transformer (DPT) and Depth Anything V2). For object detection and pose estimation, performance comparisons of YOLO11 (YOLO11n, YOLO11s, YOLO11m, YOLO11l and YOLO11x) and YOLOv8 (YOLOv8n, YOLOv8s, YOLOv8m, YOLOv8l and YOLOv8x) were made under identical hyperparameter settings among the all configurations. It was observed that YOLO11n surpassed all configurations of YOLO11 and YOLOv8 in terms of box precision and pose precision, achieving scores of 0.91 and 0.915, respectively. Conversely, YOLOv8n exhibited the highest box and pose recall scores of 0.905 and 0.925, respectively. Regarding the mean average precision at 50\% intersection over union (mAP@50), YOLO11s led all configurations with a box mAP@50 score of 0.94, while YOLOv8n achieved the highest pose mAP@50 score of 0.96. In terms of image processing speed, YOLO11n outperformed all configurations with an impressive inference speed of 2.7 ms, significantly faster than the quickest YOLOv8 configuration, YOLOv8n, which processed images in 7.8 ms. Subsequent integration of ViTs for the green fruit's pose depth estimation revealed that Depth Anything V2 outperformed Dense Prediction Transformer in 3D pose length validation, achieving the lowest Root Mean Square Error (RMSE) of 1.52 and Mean Absolute Error (MAE) of 1.28, demonstrating exceptional precision in estimating immature green fruit lengths. Integration of YOLO11 and Depth Anything Model provides a promising solution to 3D pose estimation of immature green fruits for robotic thinning applications. (YOLOv11 pose detection, YOLOv11 Pose, YOLOv11 Keypoints detection, YOLOv11 pose estimation)
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。