3 个月前

用于3D点云部件分割的十字形注意力机制

用于3D点云部件分割的十字形注意力机制

摘要

我们提出了一种深度学习方法,通过在形状集合内跨形状传播点级特征表示,实现三维形状分割。为此,我们设计了一种跨形状注意力机制,使单个形状的点级特征与其他形状的特征之间能够进行交互。该机制不仅评估点与点之间的交互程度,还调控特征在不同形状间的传播过程,从而提升分割任务中点级特征表示的准确性与一致性。此外,我们还提出了一种形状检索度量方法,用于为每个测试形状选择适合进行跨形状注意力操作的参考形状。实验结果表明,所提方法在主流的PartNet数据集上取得了当前最优的分割性能。

代码仓库

marios2019/CSN
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
3d-semantic-segmentation-on-partnetCSN
mIOU: 62.1

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
用于3D点云部件分割的十字形注意力机制 | 论文 | HyperAI超神经