arXiv:2409.06683cs.CV2024-09ECCV被引 3

利用CAD模型先验,更快更准地估计物体姿态分布。

Alignist: CAD-Informed Orientation Distribution Estimation by Fusing Shape and Correspondences

  • 结合CAD模型的形状与对应关系,构建对称性感知的姿态分布先验。
  • 在SYMSOL-I和T-Less数据集上达到基准性能,收敛速度显著提升。
  • 适合需要高效精准姿态估计的机器人抓取与路径规划场景。

物体姿态分布估计在机器人领域对路径规划和对称物体处理至关重要。现有方法多基于对比学习,在缺乏CAD模型时仅优化单一姿态估计,需大量多视角训练图像,难以在真实场景中实现。本文提出一种新方法,利用CAD模型提供的形状信息和对称性尊重的对应关系分布,指导姿态分布学习。该先验可帮助网络聚焦于多个有效姿态模式附近的尖锐分布,而非逐个优化单一模式。通过引入分布间损失,模型能更高效收敛并获得更优分布估计。在SYMSOL-I和T-Less数据集上取得当前最优结果。

原文摘要 · Abstract (English)

Object pose distribution estimation is crucial in robotics for better path planning and handling of symmetric objects. Recent distribution estimation approaches employ contrastive learning-based approaches by maximizing the likelihood of a single pose estimate in the absence of a CAD model. We propose a pose distribution estimation method leveraging symmetry respecting correspondence distributions and shape information obtained using a CAD model. Contrastive learning-based approaches require an exhaustive amount of training images from different viewpoints to learn the distribution properly, which is not possible in realistic scenarios. Instead, we propose a pipeline that can leverage correspondence distributions and shape information from the CAD model, which are later used to learn pose distributions. Besides, having access to pose distribution based on correspondences before learning pose distributions conditioned on images, can help formulate the loss between distributions. The prior knowledge of distribution also helps the network to focus on getting sharper modes instead. With the CAD prior, our approach converges much faster and learns distribution better by focusing on learning sharper distribution near all the valid modes, unlike contrastive approaches, which focus on a single mode at a time. We achieve benchmark results on SYMSOL-I and T-Less datasets.

姿态估计CAD先验分布学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。