F2G-Pose оценивает позу объектов по RGB-D за один проход, 28 кадров/с

F2G-Pose оценивает позу объектов по RGB-D за один проход, 28 кадров/с

В статье рассматривается оценка позы объекта на уровне категории по RGB-D-данным (цвет плюс глубина): нужно восстановить трёхмерный поворот и положение объекта без CAD-моделей конкретных экземпляров и без эталонных ракурсов. По описанию статьи, задача сильно неоднозначна из-за различий внутри класса, симметрий и частичной видимости объекта.

Авторы отмечают проблему существующих подходов: методы с промежуточными структурами склонны к накоплению ошибок, а диффузионные модели страдают от высокой задержки вывода и сложного отбора кандидатов после генерации. Баланс надёжности и скорости пока не найден.

В ответ предложен F2G-Pose, прямой однопроходный метод реального времени. По одному RGB-D-наблюдению за один прямой проход сети он предсказывает позу на уровне категории, метрический размер и достроенную форму объекта в системе координат камеры.

Устройство метода: плотные признаки визуальной «фундаментальной» модели переносятся на частичные облака точек; локальная геометрия кодируется токенизацией на основе DGCNN; структуру объекта в целом объединяет трансформер со смесью экспертов (Mixture-of-Experts). Совместное предсказание размера и достройка формы в системе камеры обеспечивают метрическую согласованность за счёт плотного геометрического обучающего сигнала.

Специализированная сеть обучена только на синтетических данных SOPE с разметкой поз, а замороженный backbone RADIO предобучен отдельно. По заявлению авторов, F2G-Pose достигает наилучшей точности на SOPE и ROPE при 28 FPS при выводе только позы. Также сообщается о сильной геометрии в системе камеры на HouseCat6D и переносе на реальные объекты на HANDAL. Анализ задержки, устойчивости, достройки формы и поведения кандидатов GenPose++ подтверждает, по словам авторов, эффективность и надёжность метода; эксперименты с реальным манипулированием показывают потенциал применения при сегментированных RGB-D-входах. Страница проекта, код и предобученные модели опубликованы: https://jrryzh.github.io/F2G-Pose/.

Ключевые факты

  • F2G-Pose за один прямой проход по одному RGB-D-снимку предсказывает позу на уровне категории, метрический размер и достроенную форму объекта в системе камеры.
  • Архитектура: признаки фундаментальной модели переносятся на облака точек, локальная геометрия кодируется через DGCNN, структуру объекта объединяет трансформер со смесью экспертов; backbone RADIO заморожен.
  • Сеть обучена только на синтетических данных SOPE с разметкой поз; авторы заявляют лучшую точность на SOPE и ROPE и 28 FPS при выводе только позы.
  • Дополнительно заявлены результаты на HouseCat6D (геометрия в системе камеры) и HANDAL (перенос на реальные объекты), а также эксперименты с реальным манипулированием.
  • Страница проекта, код и предобученные модели опубликованы.

Почему это важно

Оценка позы объектов без CAD-моделей конкретных экземпляров нужна там, где система встречает новые предметы известных категорий. По описанию статьи, существующие подходы упираются в компромисс: промежуточные структуры накапливают ошибки, а диффузионные модели медленны и требуют сложного отбора кандидатов. F2G-Pose заявлен как прямой однопроходный вариант, работающий со скоростью 28 FPS при выводе только позы.

Кому это важно

Исследователям компьютерного зрения и робототехники, занимающимся оценкой 6D-позы и захватом объектов, а также тем, кто сравнивает однопроходные методы с диффузионными, например GenPose++.

Как это применить

Авторы опубликовали страницу проекта, код и предобученные модели (jrryzh.github.io/F2G-Pose). Эксперименты с манипулированием в статье описаны для сегментированных RGB-D-входов, то есть для применения нужна предварительная сегментация объекта. Сеть обучена только на синтетических данных SOPE.

Можно ли доверять

Все показатели, заявления авторов из аннотации. Конкретных метрик точности и отрыва от прежних методов в тексте нет, нет и сведений об аппаратуре, на которой измерены 28 FPS, а также о скорости полного конвейера с достройкой формы. Цифры нужно проверять по полной статье и коду.

Риски и подводные камни

28 FPS относятся только к выводу позы; скорость с достройкой формы не названа. Детали реальных экспериментов с манипулированием (робот, задачи, доля успехов) не приведены. Обучение только на синтетике и зависимость от внешнего замороженного backbone RADIO означают, что качество на реальных сценах стоит проверять отдельно.