Geo6D: Geometric Constraints Learning for 6D Pose Estimation

Jianqiu Chen,Mingshan Sun,Ye Zheng,Tianpeng Bao,Zhenyu He,Donghai Li,Guoqiang Jin,Rui Zhao,Liwei Wu,Xiaoke Jiang
DOI: https://doi.org/10.48550/arxiv.2210.10959
2023-01-01
Abstract: Existing direct 6D pose estimation methods regress target 6D poses without the need for post-processing, making them effective and easy to develop. However, due to the lack of geometric constraints, the precision of these approaches remains limited, and more training data are required to achieve better performance. To this end, we propose a geometric constraint-based 6D pose estimation method (Geo6D), which establishes an explicit geometric constraint between the input and the regression target. Specifically, a Geo head is proposed to build a point-to-point relationship between the camera frame and the object frame. And additional inputs are introduced as supplementary information to facilitate the learning of the geometric correlation. The proposed Geo6D can be used as a plugin in mainstream direct 6D pose estimation methods. Extensive experimental results show that when equipped with Geo6D, the direct 6D method achieves state-of-the-art performance on multiple datasets and shows significant effectiveness with limited data volume.
What problem does this paper attempt to address?