6D-Diff: A Keypoint Diffusion Framework for 6D Object Pose Estimation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Xu, Li, Qu, Haoxuan, Cai, Yujun, Liu, Jun
Format: Preprint
Publié: 2023
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866917619894845440
author Xu, Li
Qu, Haoxuan
Cai, Yujun
Liu, Jun
author_facet Xu, Li
Qu, Haoxuan
Cai, Yujun
Liu, Jun
contents Estimating the 6D object pose from a single RGB image often involves noise and indeterminacy due to challenges such as occlusions and cluttered backgrounds. Meanwhile, diffusion models have shown appealing performance in generating high-quality images from random noise with high indeterminacy through step-by-step denoising. Inspired by their denoising capability, we propose a novel diffusion-based framework (6D-Diff) to handle the noise and indeterminacy in object pose estimation for better performance. In our framework, to establish accurate 2D-3D correspondence, we formulate 2D keypoints detection as a reverse diffusion (denoising) process. To facilitate such a denoising process, we design a Mixture-of-Cauchy-based forward diffusion process and condition the reverse process on the object features. Extensive experiments on the LM-O and YCB-V datasets demonstrate the effectiveness of our framework.
format Preprint
id arxiv_https___arxiv_org_abs_2401_00029
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle 6D-Diff: A Keypoint Diffusion Framework for 6D Object Pose Estimation
Xu, Li
Qu, Haoxuan
Cai, Yujun
Liu, Jun
Computer Vision and Pattern Recognition
Estimating the 6D object pose from a single RGB image often involves noise and indeterminacy due to challenges such as occlusions and cluttered backgrounds. Meanwhile, diffusion models have shown appealing performance in generating high-quality images from random noise with high indeterminacy through step-by-step denoising. Inspired by their denoising capability, we propose a novel diffusion-based framework (6D-Diff) to handle the noise and indeterminacy in object pose estimation for better performance. In our framework, to establish accurate 2D-3D correspondence, we formulate 2D keypoints detection as a reverse diffusion (denoising) process. To facilitate such a denoising process, we design a Mixture-of-Cauchy-based forward diffusion process and condition the reverse process on the object features. Extensive experiments on the LM-O and YCB-V datasets demonstrate the effectiveness of our framework.
title 6D-Diff: A Keypoint Diffusion Framework for 6D Object Pose Estimation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2401.00029