arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.18791cs.CV

通过扩散特征聚合实现物体姿态估计

Accurate and Efficient Object Pose Estimation via the Aggregation of Diffusion Features

Shenyang Institute of Automation, Chinese Academy of Sciences(中国科学院沈阳自动化研究所);Institutes for Robotics and Intelligent Manufacturing, Chinese Academy of Sciences(中国科学院机器人与智能制造创新研究院);University of Chinese Academy of Sciences(中国科学院大学);Oosto(奥斯特公司)

Tianfu Wang, Guosheng Hu, Hongguang Wang

AI总结:

本文针对物体姿态估计在未见物体上泛化差的问题,提出利用 Stable Diffusion 等扩散模型特征,并通过三种聚合不同粒度特征的架构,在 LM、O-LM 和 T-LESS 上显著超越现有方法。

摘要:

从图像中估计物体姿态是三维场景理解中的一项关键任务,近期方法在超大规模基准上已展现出有前景的结果。然而,这些方法在处理未见过的物体时会出现显著的性能下降。我们认为这源于图像特征的泛化能力有限。为解决该问题,我们对扩散模型(例如 Stable Diffusion)的特征进行了深入分析,这些特征在建模未见过的物体方面具有巨大潜力。基于该分析,我们创新性地将这些扩散特征引入物体姿态估计。为实现这一目标,我们提出了三种不同的架构,能够有效捕获并聚合不同粒度的扩散特征,从而大幅提升物体姿态估计的泛化能力。我们的方法在三个常用基准数据集 LM、O-LM 和 T-LESS 上以相当大的优势超越了当前最先进的方法。特别是,在未见过的物体上,我们的方法取得了比此前最佳方法更高的准确率:在 Unseen LM 上为 97.9% 对 93.5%,在 Unseen O-LM 上为 85.9% 对 76.3%,显示出我们方法强大的泛化能力。我们的代码已发布在 https://github.com/Tianfu18/diff-feats-pose。

英文摘要:

Estimating the pose of objects from images is a crucial task of 3D scene understanding, and recent approaches have shown promising results on very large benchmarks. However, these methods experience a significant performance drop when dealing with unseen objects. To address this problem, we have an in-depth analysis on the features of diffusion models, e.g. Stable Diffusion, which hold substantial potential for modeling unseen objects. Based on this analysis, we then innovatively introduce these diffusion features for object pose estimation. To verify the efficacy of diffusion features for object pose estimation, we propose three distinct architectures (vanilla, nonlinear, and context-aware weight aggregations) that capture and aggregate diffusion features for comparative analysis. To achieve an efficient feature aggregation, we propose a confidence adaptive aggregation network that automatically selects the discriminative features rather than uses all the features, achieving a better speed-and-accuracy trade-off. In particular, our confidence adaptive aggregation network achieves higher accuracy than the previous best arts on unseen objects: 97.7% vs. 93.5% on Unseen LM, 85.5% vs. 76.3% on Unseen O-LM, showing the strong generalizability of our method. On the large-scale BOP benchmark, our method also provides measurable gains, with an average recall of 58.3 compared to 57.9 previously. In addition, CAA reduces computational cost by 1.3-1.5 compared to the CWA variant while maintaining comparable accuracy. Furthermore, CAA reaches real-time performance, achieving over 68 FPS and offering a substantially improved accuracy-efficiency trade-off.

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

↑