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 означают, что качество на реальных сценах стоит проверять отдельно.