3 个月前

HybridPose:基于混合表示的6D物体位姿估计

HybridPose:基于混合表示的6D物体位姿估计

摘要

我们提出了一种名为 HybridPose 的新型 6D 物体位姿估计方法。HybridPose 采用一种混合中间表示(hybrid intermediate representation),用于在输入图像中表达多种几何信息,包括关键点、边缘向量以及对称对应关系。相较于单一表示方式,该混合表示机制使得位姿回归能够在某一类预测结果不准确时(例如因遮挡导致),依然能够利用更多样化的特征进行推断。HybridPose 所采用的不同中间表示均可由同一个简洁的神经网络统一预测,且预测结果中的异常值可通过一个鲁棒的回归模块进行有效过滤。与当前最先进的位姿估计方法相比,HybridPose 在运行速度和精度方面均表现出相当的性能。例如,在 Occlusion Linemod 数据集上,该方法实现了 30 fps 的预测速度,并达到 47.5% 的平均 ADD(-S) 精度,达到了当前最优水平。HybridPose 的开源实现已发布于 https://github.com/chensong1995/HybridPose。

代码仓库

chensong1995/HybridPose
官方
pytorch
GitHub 中提及
hz-ants/HybridPose
pytorch
GitHub 中提及
chronoshell/copyydscsv
pytorch
GitHub 中提及

基准测试

基准方法指标
6d-pose-estimation-on-linemodHybridPose
Accuracy (ADD): 94.5%
Mean ADD: 91.3
6d-pose-estimation-using-rgb-on-occlusionHybridPose
Mean ADD: 47.5

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
HybridPose:基于混合表示的6D物体位姿估计 | 论文 | HyperAI超神经