Learning Point Cloud Representations with Pose Continuity for Depth-Based Category-Level 6D Object Pose Estimation
2508.14358v1
cs.CV, cs.AI, cs.RO
2025-08-22
Авторы:
Zhujun Li, Shuo Zhang, Ioannis Stamos
Резюме на русском
## Контекст
Category-level 6D object pose estimation является важной задачей в области зрения компьютера, нацеленной на определение 6D-позы и 3D-размеров объектов в представлении данных глубины. Данная задача находит применение в ассистированном робототводе, виртуальной реальности и других областях. Однако, существующие подходы, основанные на простых 6D-позных сигналах, часто страдают от несогласованности предсказаний и ограниченности в области новых поз, что снижает качество решения. Мотивирует нашу работу необходимость в улучшении точности и устойчивости предсказаний для универсального использования.
## Метод
Мы предлагаем HRC-Pose, новую методику для выделения представлений точечных облаков, сохраняющих пределы 6D-позы. Наша методика основывается на контрастном обучении, который позволяет учитывать связанность между 6D-позами. HRC-Pose разделяет 6D-позу на компоненты поворота и перемещения, которые обрабатываются отдельно. Мы предлагаем 6D pose-aware hierarchical ranking scheme, который использует 6D-позы для сравнения точечных облаков из нескольких категорий. Для позиционирования, мы разработали модули для обработки rotation-aware и translation-aware embeddings, улучшая стабильность и точность.
## Результаты
Мы проверили HRC-Pose на двух наборах данных REAL275 и CAMERA25. Выборка REAL275 включает реальные сцены, а CAMERA25 — синтетические. Наши эксперименты показали, что HRC-Pose выдает значительно улучшенную точность во всех категориях в сравнении с состоянием искусства, специально для глубинных данных. Мы также показали, что методика работает в реальном времени, что демонстрирует ее применность в реальных системах.
## Значимость
Методика HRC-Pose может применяться в различных областях, таких как ассистированный робот, системы AR/VR и автоматизированная сборка. Она предлагает более точные и устойчивые предсказания 6D-позы в сравнении с состоянием искусства. Благодаря своей эффективности, она позволяет улучшить качество работы в реальных условиях.
## Выводы
Мы доказали, что HRC-Pose позволяет улучшить точность 6D-позы в depth-based category-level estimation. Будущие исследования будут направлены на улучшение сложности модели и расширение ее применения к более широкой области 6D-поз.
Abstract
Category-level object pose estimation aims to predict the 6D pose and 3D size
of objects within given categories. Existing approaches for this task rely
solely on 6D poses as supervisory signals without explicitly capturing the
intrinsic continuity of poses, leading to inconsistencies in predictions and
reduced generalization to unseen poses. To address this limitation, we propose
HRC-Pose, a novel depth-only framework for category-level object pose
estimation, which leverages contrastive learning to learn point cloud
representations that preserve the continuity of 6D poses. HRC-Pose decouples
object pose into rotation and translation components, which are separately
encoded and leveraged throughout the network. Specifically, we introduce a
contrastive learning strategy for multi-task, multi-category scenarios based on
our 6D pose-aware hierarchical ranking scheme, which contrasts point clouds
from multiple categories by considering rotational and translational
differences as well as categorical information. We further design pose
estimation modules that separately process the learned rotation-aware and
translation-aware embeddings. Our experiments demonstrate that HRC-Pose
successfully learns continuous feature spaces. Results on REAL275 and CAMERA25
benchmarks show that our method consistently outperforms existing depth-only
state-of-the-art methods and runs in real-time, demonstrating its effectiveness
and potential for real-world applications. Our code is at
https://github.com/zhujunli1993/HRC-Pose.
Ссылки и действия
Дополнительные ресурсы: