17 天前

SO-Net:用于点云分析的自组织网络

SO-Net:用于点云分析的自组织网络

摘要

本文提出SO-Net,一种适用于无序点云的深度学习排列不变架构。SO-Net通过构建自组织映射(Self-Organizing Map, SOM)来建模点云的空间分布。基于该SOM结构,SO-Net在单个点与SOM节点上进行分层特征提取,并最终以单一特征向量表示输入点云。通过执行点到节点的k近邻搜索,可系统性地调节网络的感受野。在点云重建、分类、物体部件分割及形状检索等识别任务中,所提出的网络性能与现有最先进方法相当或更优。此外,由于该架构具有良好的并行性和结构简洁性,其训练速度显著快于现有的点云识别网络。项目代码已开源,可在以下网址获取:https://github.com/lijx10/SO-Net

代码仓库

LONG-9621/SO-Net
pytorch
GitHub 中提及
lijx10/SO-Net
官方
pytorch
GitHub 中提及
donnyruixu/pc-elm-ae
pytorch
GitHub 中提及

基准测试

基准方法指标
3d-part-segmentation-on-intraSO-Net
DSC (A): 88.76
DSC (V): 97.09
IoU (A): 81.40
IoU (V): 94.46
3d-part-segmentation-on-shapenet-partSO-Net
Instance Average IoU: 84.9
3d-point-cloud-classification-on-intraSO-Net
F1 score (5-fold): 0.868
3d-point-cloud-classification-on-modelnet40SO-Net
Overall Accuracy: 90.9
3d-point-cloud-linear-classification-onSO-Net
Overall Accuracy: 87.5

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
SO-Net:用于点云分析的自组织网络 | 论文 | HyperAI超神经