CDHN: Cross-domain hallucination network for 3D keypoints estimation
This paper presents a novel method to estimate sparse 3D keypoints from single-view RGB images. Our network is trained in two steps using a knowledge distillation framework. In the first step, the teacher is trained to extract 3D features from point cloud data, which are used in combination with 2D features to estimate the 3D keypoints. In the second step, the teacher teaches the student module to hallucinate the 3D features from RGB images that are similar to those extracted from the point clouds. This procedure helps the network during inference to extract 2D and 3D features directly from images, without requiring point clouds as input. Moreover, the network also predicts a confidence score for every keypoint, which is used to select the valid ones from a set of N predicted keypoints. This allows the prediction of different number of keypoints depending on the object’s geometry. We use the estimated keypoints for computing the relative pose between two views of an object. The results are compared with those of KP-Net and StarMap , which are the state-of-the-art for estimating 3D keypoints from a single-view RGB image. The average angular distance error of our approach (5.94°) is 8.46° and 55.26° lower than that of KP-Net (14.40°) and StarMap (61.20°), respectively. • We estimate keypoints from RGB images by leveraging information learnt from 3D data. • Our approach is flexible that can predict geometry based number of keypoints. • It can be trained for various categories simultaneously. • It is capable of estimating keypoints of self-occluded parts of the objects. • The estimated keypoints can be used for relative pose estimation.
Authors
- Pietro Morerio (ORCID: https://orcid.org/0000-0001-5259-1496)
- Matteo Taiana (ORCID: https://orcid.org/0000-0003-1759-2447)
- Milind G. Padalkar (ORCID: https://orcid.org/0000-0002-0342-5448)
- Alessio Del Bue (ORCID: https://orcid.org/0000-0002-2262-4872)
- Mohammad Zohaib (ORCID: https://orcid.org/0000-0003-2259-4121)
Publication Details
- Journal
- Pattern Recognition
- Published
- 2024-11-15
- DOI
- https://doi.org/10.1016/j.patcog.2024.111188
- Citations
- 1
- Primary Topic
- Advanced Vision and Imaging
- Type
- article
- Field-Weighted Citation Impact
- 0.25
Funders
- European Commission
- Istituto Italiano di Tecnologia