4 个月前

SSD-6D:让基于RGB的3D检测和6D姿态估计再次强大

SSD-6D:让基于RGB的3D检测和6D姿态估计再次强大

摘要

我们提出了一种新颖的方法,可以从单次拍摄的RGB数据中检测3D模型实例并估计其6D姿态。为此,我们将流行的SSD范式扩展到覆盖完整的6D姿态空间,并仅使用合成模型数据进行训练。我们的方法在多个具有挑战性的数据集上与当前最先进的利用RGB-D数据的方法竞争或超越。此外,我们的方法以大约10赫兹的速度产生这些结果,比相关方法快数倍。为了便于复现,我们公开发布了训练好的网络和检测代码。

代码仓库

基准测试

基准方法指标
6d-pose-estimation-on-linemodSSD-6D
Mean ADD: 76.3
Mean IoU: 99.4
6d-pose-estimation-on-occlusionSSD-6D
MAP: 0.38
6d-pose-estimation-using-rgbd-on-linemodSSD-6D
Mean ADD: 90.9
Mean IoU: 96.5
6d-pose-estimation-using-rgbd-on-tejaniSSD-6D
IoU-2D: 0.988
IoU-3D: 0.963
VSS-2D: 0.724
VSS-3D: 0.854

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
SSD-6D:让基于RGB的3D检测和6D姿态估计再次强大 | 论文 | HyperAI超神经