SAR-Net: Shape Alignment and Recovery Network for Category-level 6D Object Pose and Size Estimation


Fudan University     IDL, Baidu Reasearch





Abstract

Given a single scene image, this paper proposes a method of Category-level 6D Object Pose and Size Estimation (COPSE) from the point cloud of the target object, without external real pose-annotated training data. Specifically, beyond the visual cues in RGB images, we rely on the shape information predominately from the depth (D) channel. The key idea is to explore the shape alignment of each instance against its corresponding category-level template shape, and the symmetric correspondence of each object category for estimating a coarse 3D object shape. Our framework deforms the point cloud of the category-level template shape to align the observed instance point cloud for implicitly representing its 3D rotation. Then we model the symmetric correspondence by predicting symmetric point cloud from the partially observed point cloud. The concatenation of the observed point cloud and symmetric one reconstructs a coarse object shape, thus facilitating object center (3D translation) and 3D size estimation. Extensive experiments on the category-level NOCS benchmark demonstrate that our lightweight model still competes with state-of-the-art approaches that require labeled real-world images. We also deploy our approach to a physical Baxter robot to perform grasping tasks on unseen but category-known instances, and the results further validate the efficacy of our proposed model.




Network Architecture




Paper and Code

H. Lin, Z. Liu, C. Cheang, Y. Fu, G. Guo, X. Xue

SAR-Net: Shape Alignment and Recovery Network for Category-level 6D Object Pose and Size Estimation

CVPR 2022.

[arXiv]     [GitHub]    



Videos on bilibili

Videos on YouTube




More demos












Quantitative Results on NOCS REAL275 dataset




Quantitative Results on addtional real-world scenarios




Acknowledgements

Xiangyang Xue and Yanwei Fu are the corresponding authors. This work was supported in part by NSFC under Grant (No. 62076067), Shanghai Municipal Science and Technology Major Project (2018SHZDZX01). The website is modified from this template.