arXiv · 2412.11420
Category Level 6D Object Pose Estimation from a Single RGB Image using Diffusion
Abstract
Estimating the 6D pose and 3D size of an object from an image is a fundamental task in computer vision. Most current approaches are restricted to specific instances with known models or require ground truth depth information or point cloud captures from LIDAR. We tackle the harder problem of pose estimation for category-level objects from a single RGB image. We propose a novel solution that eliminates the need for specific object models or depth information. Our method utilises score-based diffusion models to generate object pose hypotheses to model the distribution of possible poses for the object. Unlike previous methods that rely on costly trained likelihood estimators to remove outliers before pose aggregation using mean pooling, we introduce a simpler approach using Mean Shift to estimate the mode of the distribution as the final pose estimate. Our approach outperforms the current state-of-the-art on the REAL275 dataset by a significant margin.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Adam Bethell, Ravi Garg, Ian Reid. 2024-12-16. Category Level 6D Object Pose Estimation from a Single RGB Image using Diffusion. https://arxiv.org/abs/2412.11420
Cite the original work for its findings. Save a collection to share your selection of sources.