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

基于加权向量式关键点投票的深度融合Transformer网络用于鲁棒6D物体姿态估计

Deep Fusion Transformer Network with Weighted Vector-Wise Keypoints Voting for Robust 6D Object Pose Estimation

The Hong Kong Polytechnic University(香港理工大学);The Chinese University of Hong Kong(香港中文大学);University of Waterloo(滑铁卢大学)

Jun Zhou, Kai Chen, Linlin Xu, Qi Dou, Jing Qin

AI总结:

本文提出深度融合Transformer模块和加权向量式投票算法,通过跨模态语义关联建模与非迭代全局优化实现精确3D关键点定位,在四个基准上以较大幅度超越现有方法。

摘要:

从单张RGBD图像进行6D物体姿态估计的一个关键挑战是如何高效地整合颜色和深度这两种不同模态。本文通过提出一种新颖的深度融合Transformer(DFTr)模块来解决该问题,该模块能够聚合跨模态特征以提升姿态估计性能。与现有融合方法不同,所提出的DFTr通过利用模态间的语义相似性,能够更好地建模跨模态语义相关性,从而使来自不同模态的全局增强特征得到更好的整合,以改善信息提取。此外,为了进一步提高鲁棒性和效率,我们引入了一种新颖的加权向量式投票算法,该算法采用非迭代的全局优化策略来实现精确的3D关键点定位,同时实现近实时推理。大量实验表明了我们所提出的3D关键点投票算法的有效性和强泛化能力。在四个广泛使用的基准上的结果也证明,我们的方法以较大幅度优于当前最先进的方法。

英文摘要:

One critical challenge in 6D object pose estimation from a single RGBD image is efficient integration of two different modalities, i.e., color and depth. In this work, we tackle this problem by a novel Deep Fusion Transformer~(DFTr) block that can aggregate cross-modality features for improving pose estimation. Unlike existing fusion methods, the proposed DFTr can better model cross-modality semantic correlation by leveraging their semantic similarity, such that globally enhanced features from different modalities can be better integrated for improved information extraction. Moreover, to further improve robustness and efficiency, we introduce a novel weighted vector-wise voting algorithm that employs a non-iterative global optimization strategy for precise 3D keypoint localization while achieving near real-time inference. Extensive experiments show the effectiveness and strong generalization capability of our proposed 3D keypoint voting algorithm. Results on four widely used benchmarks also demonstrate that our method outperforms the state-of-the-art methods by large margins.

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

↑