3 个月前

自顶向下优于自底向上:3D实例分割

自顶向下优于自底向上:3D实例分割

摘要

大多数3D实例分割方法采用自底向上的策略,通常包含资源消耗较大的后处理步骤。在点云分组方面,自底向上的方法依赖于关于物体的先验假设,以超参数形式表达,这些超参数具有领域特定性,需进行精细调参。相比之下,本文提出TD3D:一种开创性的无聚类、全卷积且完全数据驱动的端到端训练方法,是首个在3D领域中性能超越自底向上方法的自顶向下方法。该方法具有简洁的处理流程,在标准室内基准数据集ScanNet v2、其扩展版本ScanNet200以及S3DIS上均展现出卓越的精度与泛化能力,同时在航拍数据集STPLS3D上也表现优异。此外,与当前最先进的基于分组的方法相比,本方法在推理速度上显著更快:其主干版本在保持更高精度的前提下,比最准确的自底向上方法快1.9倍;而更快速的变体则在实现当前最先进精度的同时,推理速度提升了2.6倍。代码已开源,地址为:https://github.com/SamsungLabs/td3d。

基准测试

基准方法指标
3d-instance-segmentation-on-s3disTD3D
AP@50: 70.4
mAP: 58.1
3d-instance-segmentation-on-scannetv2TD3D
mAP: 48.9
mAP @ 50: 75.1
mAP@25: 87.5
3d-instance-segmentation-on-stpls3dTD3D
AP: 54.3
AP50: 69.8

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
自顶向下优于自底向上:3D实例分割 | 论文 | HyperAI超神经