4 个月前

3D-MPA:用于3D语义实例分割的多提案聚合

3D-MPA:用于3D语义实例分割的多提案聚合

摘要

我们提出了3D-MPA,一种用于3D点云实例分割的方法。对于给定的输入点云,我们提出了一种以对象为中心的方法,其中每个点都会为其所属的对象中心投票。我们从预测的对象中心中采样生成对象提案。然后,我们从投票给同一对象中心的分组点特征中学习提案特征。图卷积网络引入了提案之间的关系,除了低级别的点特征外,还提供了高级别的特征学习。每个提案包含一个语义标签、一组定义前景-背景掩码的相关点、一个对象得分以及聚合特征。以往的工作通常通过对提案执行非极大值抑制(NMS)来获得最终的对象检测或语义实例。然而,NMS可能会丢弃潜在正确的预测结果。相比之下,我们的方法保留所有提案,并根据学习到的聚合特征将它们分组在一起。我们展示了在ScanNetV2基准和S3DIS数据集上,通过分组提案的方法在3D对象检测和语义实例分割任务中优于NMS,并且超过了之前最先进的方法。

代码仓库

francisengelmann/3D-MPA
GitHub 中提及

基准测试

基准方法指标
3d-instance-segmentation-on-s3dis3D-MPA
mPrec: 66.7
mRec: 64.1
3d-instance-segmentation-on-scannetv23D-MPA
mAP: 35.3
mAP @ 50: 59.1
mRec: 61.1
3d-object-detection-on-scannetv23D-MPA
mAP@0.25: 64.2
mAP@0.5: 49.2
3d-semantic-instance-segmentation-on-13D-MPA
mAP@0.50: 61.1

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
3D-MPA:用于3D语义实例分割的多提案聚合 | 论文 | HyperAI超神经