Diff9D: Diffusion-Based Domain-Generalized Category-Level 9-DoF Object Pose Estimation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Liu, Jian, Sun, Wei, Yang, Hui, Deng, Pengchao, Liu, Chongpei, Sebe, Nicu, Rahmani, Hossein, Mian, Ajmal
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866915199623102464
author Liu, Jian
Sun, Wei
Yang, Hui
Deng, Pengchao
Liu, Chongpei
Sebe, Nicu
Rahmani, Hossein
Mian, Ajmal
author_facet Liu, Jian
Sun, Wei
Yang, Hui
Deng, Pengchao
Liu, Chongpei
Sebe, Nicu
Rahmani, Hossein
Mian, Ajmal
contents Nine-degrees-of-freedom (9-DoF) object pose and size estimation is crucial for enabling augmented reality and robotic manipulation. Category-level methods have received extensive research attention due to their potential for generalization to intra-class unknown objects. However, these methods require manual collection and labeling of large-scale real-world training data. To address this problem, we introduce a diffusion-based paradigm for domain-generalized category-level 9-DoF object pose estimation. Our motivation is to leverage the latent generalization ability of the diffusion model to address the domain generalization challenge in object pose estimation. This entails training the model exclusively on rendered synthetic data to achieve generalization to real-world scenes. We propose an effective diffusion model to redefine 9-DoF object pose estimation from a generative perspective. Our model does not require any 3D shape priors during training or inference. By employing the Denoising Diffusion Implicit Model, we demonstrate that the reverse diffusion process can be executed in as few as 3 steps, achieving near real-time performance. Finally, we design a robotic grasping system comprising both hardware and software components. Through comprehensive experiments on two benchmark datasets and the real-world robotic system, we show that our method achieves state-of-the-art domain generalization performance. Our code will be made public at https://github.com/CNJianLiu/Diff9D.
format Preprint
id arxiv_https___arxiv_org_abs_2502_02525
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Diff9D: Diffusion-Based Domain-Generalized Category-Level 9-DoF Object Pose Estimation
Liu, Jian
Sun, Wei
Yang, Hui
Deng, Pengchao
Liu, Chongpei
Sebe, Nicu
Rahmani, Hossein
Mian, Ajmal
Computer Vision and Pattern Recognition
Robotics
Nine-degrees-of-freedom (9-DoF) object pose and size estimation is crucial for enabling augmented reality and robotic manipulation. Category-level methods have received extensive research attention due to their potential for generalization to intra-class unknown objects. However, these methods require manual collection and labeling of large-scale real-world training data. To address this problem, we introduce a diffusion-based paradigm for domain-generalized category-level 9-DoF object pose estimation. Our motivation is to leverage the latent generalization ability of the diffusion model to address the domain generalization challenge in object pose estimation. This entails training the model exclusively on rendered synthetic data to achieve generalization to real-world scenes. We propose an effective diffusion model to redefine 9-DoF object pose estimation from a generative perspective. Our model does not require any 3D shape priors during training or inference. By employing the Denoising Diffusion Implicit Model, we demonstrate that the reverse diffusion process can be executed in as few as 3 steps, achieving near real-time performance. Finally, we design a robotic grasping system comprising both hardware and software components. Through comprehensive experiments on two benchmark datasets and the real-world robotic system, we show that our method achieves state-of-the-art domain generalization performance. Our code will be made public at https://github.com/CNJianLiu/Diff9D.
title Diff9D: Diffusion-Based Domain-Generalized Category-Level 9-DoF Object Pose Estimation
topic Computer Vision and Pattern Recognition
Robotics
url https://arxiv.org/abs/2502.02525