HyperAIHyperAI

Command Palette

Search for a command to run...

用于点云三维实例分割的 SoftGroup

Thang Vu Kookhoi Kim Tung M. Luu Xuan Thanh Nguyen Chang D. Yoo

摘要

现有的最先进3D实例分割方法通常采用“语义分割后分组”的范式。在进行语义分割时,模型会做出硬性预测,即每个点仅被分配至单一类别。然而,这种硬性决策所引入的误差会传播至后续的分组阶段,导致两个主要问题:(1) 预测实例与真实标注之间的重叠率较低;(2) 产生大量假阳性实例。为解决上述问题,本文提出一种名为SoftGroup的3D实例分割方法,该方法采用自底向上的软分组策略,随后通过自顶向下的精炼机制进行优化。SoftGroup允许每个点同时关联多个类别,从而缓解因语义预测错误带来的负面影响,并通过学习将假阳性实例识别为背景类别,有效抑制其产生。在多个数据集及多种评估指标下的实验结果表明,SoftGroup具有显著优越性:在ScanNet v2隐藏测试集上,其AP_50指标较最强的先前方法提升6.2%;在S3DIS Area 5数据集上,提升幅度达6.8%。此外,SoftGroup具有较高的推理效率,在单张Titan X显卡上处理每帧扫描仅需345毫秒。本文提供的源代码与训练模型已公开,可访问GitHub仓库:\url{https://github.com/thangvubk/SoftGroup.git}


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供