3D Equivariant Pose Regression via Direct Wigner-D Harmonics Prediction

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Lee, Jongmin, Cho, Minsu
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866913570261827584
author Lee, Jongmin
Cho, Minsu
author_facet Lee, Jongmin
Cho, Minsu
contents Determining the 3D orientations of an object in an image, known as single-image pose estimation, is a crucial task in 3D vision applications. Existing methods typically learn 3D rotations parametrized in the spatial domain using Euler angles or quaternions, but these representations often introduce discontinuities and singularities. SO(3)-equivariant networks enable the structured capture of pose patterns with data-efficient learning, but the parametrizations in spatial domain are incompatible with their architecture, particularly spherical CNNs, which operate in the frequency domain to enhance computational efficiency. To overcome these issues, we propose a frequency-domain approach that directly predicts Wigner-D coefficients for 3D rotation regression, aligning with the operations of spherical CNNs. Our SO(3)-equivariant pose harmonics predictor overcomes the limitations of spatial parameterizations, ensuring consistent pose estimation under arbitrary rotations. Trained with a frequency-domain regression loss, our method achieves state-of-the-art results on benchmarks such as ModelNet10-SO(3) and PASCAL3D+, with significant improvements in accuracy, robustness, and data efficiency.
format Preprint
id arxiv_https___arxiv_org_abs_2411_00543
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle 3D Equivariant Pose Regression via Direct Wigner-D Harmonics Prediction
Lee, Jongmin
Cho, Minsu
Computer Vision and Pattern Recognition
Machine Learning
Robotics
Image and Video Processing
Determining the 3D orientations of an object in an image, known as single-image pose estimation, is a crucial task in 3D vision applications. Existing methods typically learn 3D rotations parametrized in the spatial domain using Euler angles or quaternions, but these representations often introduce discontinuities and singularities. SO(3)-equivariant networks enable the structured capture of pose patterns with data-efficient learning, but the parametrizations in spatial domain are incompatible with their architecture, particularly spherical CNNs, which operate in the frequency domain to enhance computational efficiency. To overcome these issues, we propose a frequency-domain approach that directly predicts Wigner-D coefficients for 3D rotation regression, aligning with the operations of spherical CNNs. Our SO(3)-equivariant pose harmonics predictor overcomes the limitations of spatial parameterizations, ensuring consistent pose estimation under arbitrary rotations. Trained with a frequency-domain regression loss, our method achieves state-of-the-art results on benchmarks such as ModelNet10-SO(3) and PASCAL3D+, with significant improvements in accuracy, robustness, and data efficiency.
title 3D Equivariant Pose Regression via Direct Wigner-D Harmonics Prediction
topic Computer Vision and Pattern Recognition
Machine Learning
Robotics
Image and Video Processing
url https://arxiv.org/abs/2411.00543