arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2605.08059cs.CVcs.RO

通过RGB-D残差神经网络进行6D位姿估计

6D Pose Estimation via Keypoint Heatmap Regression with RGB-D Residual Neural Networks

Politecnico di Torino(托里诺理工学院)

Ismail Aljosevic, Amir Masoud Almasi, Ana Parovic, Ashkan Shafiei

AI总结:

本文提出基于关键点热图回归的模块化框架,结合YOLOv10m进行目标检测与ResNet18预测RGB图像的2D热图,利用PnP RANSAC算法估计6D位姿,并通过RGB-D交叉融合架构提升性能,最终在LINEMOD数据集上达到92.41%的准确率。

摘要:

本文提出了一种基于关键点热图回归的模块化框架,用于6D位姿估计。我们的方法结合YOLOv10m进行目标检测,并使用基于ResNet18的网络从RGB图像预测2D热图。从这些热图中提取的关键点通过PnP RANSAC算法估计6D物体位姿。我们比较了不同的关键点选择策略,以评估其对位姿精度的影响。此外,我们通过交叉融合架构将深度数据纳入基线中,从而在多个阶段实现RGB和深度特征的交互。我们还探索了通用训练改进,如实验不同的激活函数和学习率调度策略以提高模型性能。我们的最佳RGB-only模型在LINEMOD数据集上达到了84.50%的平均ADD准确率,而RGB-D融合模型达到了92.41%的准确率。代码可在https://github.com/ameermasood/HeatNet上获得。

英文摘要:

In this paper, we propose a modular framework for 6D pose estimation based on keypoint heatmap regression. Our approach combines YOLOv10m for object detection with a ResNet18-based network that predicts 2D heatmaps from RGB images. Keypoints extracted from these heatmaps are used to estimate the 6D object pose via the PnP RANSAC algorithm. We compare different keypoint selection strategies to assess their impact on pose accuracy. Additionally, we extend the baseline by incorporating depth data using a cross-fusion architecture, which enables interaction between RGB and depth features at multiple stages. We further explore general training improvements, such as experimenting with activation functions and learning rate scheduling strategies to improve model performance. Our best RGB-only model achieved a mean ADD-based accuracy of 84.50%, while the RGB-D fusion model reached 92.41% on the LINEMOD dataset. The code is available at https://github.com/ameermasood/HeatNet.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑