不训练也能准且快地估计新物体的6D姿态。
Accurate and efficient zero-shot 6D pose estimation with frozen foundation models
- 用预训练视觉与几何模型,零样本直接推断姿态。
- 比前代快8倍,准确率提升5%,用多分割模型再增8%。
- 适合机器人、AR等需快速适配新物体的场景。
从RGBD数据中估计物体的6D姿态是计算机视觉的基础问题,广泛应用于机器人和增强现实。关键挑战是如何对训练中未见的新物体实现良好泛化。现有方法通常依赖大规模定制化合成数据进行任务特异性训练,计算开销巨大。我们提出FreeZeV2——无需训练的方法,通过利用在无关数据上预训练的视觉与几何基础模型,实现对未见物体的强泛化能力。其改进包括:(i) 稀疏特征提取策略,降低推理计算量而不损失精度;(ii) 特征感知评分机制,提升RANSAC注册中的姿态选择与候选排序;(iii) 模块化设计,支持实例分割模型集成,增强对分割误差的鲁棒性。在BOP基准七个核心数据集上,FreeZeV2创下新纪录。使用相同分割掩码时,相比FreeZe提速8倍,准确率提升5%;使用分割模型集成时,准确率再增8%,仍比FreeZe快2.5倍。FreeZeV2荣获2024年BOP挑战赛最佳整体方法奖。
原文摘要 · Abstract (English)
Estimating the 6D pose of objects from RGBD data is a fundamental problem in computer vision, with applications in robotics and augmented reality. A key challenge is achieving generalization to novel objects that were not seen during training. Most existing approaches address this by scaling up training on synthetic data tailored to the task, a process that demands substantial computational resources. But is task-specific training really necessary for accurate and efficient 6D pose estimation of novel objects? To answer No!, we introduce FreeZeV2, the second generation of FreeZe: a training-free method that achieves strong generalization to unseen objects by leveraging geometric and vision foundation models pre-trained on unrelated data. FreeZeV2 improves both accuracy and efficiency over FreeZe through three key contributions: (i) a sparse feature extraction strategy that reduces inference-time computation without sacrificing accuracy; (ii) a feature-aware scoring mechanism that improves both pose selection during RANSAC-based 3D registration and the final ranking of pose candidates; and (iii) a modular design that supports ensembles of instance segmentation models, increasing robustness to segmentation masks errors. We evaluate FreeZeV2 on the seven core datasets of the BOP Benchmark, where it establishes a new state-of-the-art in 6D pose estimation of unseen objects. When using the same segmentation masks, FreeZeV2 achieves a remarkable 8x speedup over FreeZe while also improving accuracy by 5%. When using ensembles of segmentation models, FreeZeV2 gains an additional 8% in accuracy while still running 2.5x faster than FreeZe. FreeZeV2 was awarded Best Overall Method at the BOP Challenge 2024.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。