17 天前

点云上的动态图CNN

点云上的动态图CNN

摘要

点云为计算机图形学中的诸多应用提供了灵活的几何表示方式,同时也是大多数三维数据采集设备的原始输出形式。尽管在图形学与视觉领域中,针对点云的手工设计特征早已被提出,但近年来卷积神经网络(CNN)在图像分析中取得的巨大成功,提示我们有必要将CNN的洞察迁移至点云领域。由于点云本身缺乏拓扑信息,因此设计一种能够恢复拓扑结构的模型,将显著增强点云的表示能力。为此,我们提出了一种新型神经网络模块——EdgeConv,该模块适用于基于CNN的点云高层任务,如分类与分割。EdgeConv在每一层网络中动态计算的图结构上进行操作,具有可微性,可无缝集成到现有网络架构中。与现有在外部空间中操作或独立处理每个点的模块相比,EdgeConv具备若干优越特性:它能够融合局部邻域信息;可通过堆叠多层以学习全局形状特征;在多层系统中,特征空间中的相似性可捕捉原始嵌入空间中潜在长距离的语义特性。我们在标准基准数据集(包括ModelNet40、ShapeNetPart和S3DIS)上验证了所提模型的性能。

代码仓库

yossilevii100/critical_points2
pytorch
GitHub 中提及
bupt-gamma/gammagl
tf
GitHub 中提及
Squanch-U/DGCNN
mindspore
GitHub 中提及
brent-murray/tr3d_pointaugdgcnn
pytorch
GitHub 中提及
cy-xu/spatially_aware_ai
pytorch
GitHub 中提及
AnTao97/dgcnn.pytorch
pytorch
GitHub 中提及
nnn911/MLSI
pytorch
GitHub 中提及
lingzhang1/ContrastNet
tf
GitHub 中提及
hansen7/NRS_3D
pytorch
GitHub 中提及
lingzhang1/dgcnn_sampling
tf
GitHub 中提及
vinits5/learning3d
pytorch
GitHub 中提及
hqucms/ParticleNet
tf
GitHub 中提及
AmitBracha/GIP_project
tf
GitHub 中提及
yossilevii100/refocusing
pytorch
GitHub 中提及
lingzhang1/dgcnn_v2
tf
GitHub 中提及
af13s/dgcnn-amino
tf
GitHub 中提及
JingfeiHuang/DGCNN-Paddle
paddle
GitHub 中提及
WangYueFt/dgcnn
官方
tf
GitHub 中提及

基准测试

基准方法指标
3d-part-segmentation-on-shapenet-partDGCNN
Instance Average IoU: 85.2
3d-point-cloud-classification-on-intraDGCNN
F1 score (5-fold): 0.738
3d-point-cloud-classification-on-modelnet40DGCNN
Mean Accuracy: 90.2
Number of params: 1.81M
Overall Accuracy: 92.9
3d-point-cloud-classification-on-modelnet40-cDGCNN
Error Rate: 0.259
3d-point-cloud-classification-on-scanobjectnnDGCNN
Mean Accuracy: 73.6
OBJ-BG (OA): 82.8
OBJ-ONLY (OA): 86.2
Overall Accuracy: 78.1
few-shot-3d-point-cloud-classification-on-1DGCNN
Overall Accuracy: 31.6
Standard Deviation: 9.0
few-shot-3d-point-cloud-classification-on-2DGCNN
Overall Accuracy: 40.8
Standard Deviation: 14.6
few-shot-3d-point-cloud-classification-on-3DGCNN
Overall Accuracy: 19.85
Standard Deviation: 6.5
few-shot-3d-point-cloud-classification-on-4DGCNN
Overall Accuracy: 16.9
Standard Deviation: 1.5
point-cloud-classification-on-pointcloud-cDGCNN
mean Corruption Error (mCE): 1.000
point-cloud-segmentation-on-pointcloud-cDGCNN
mean Corruption Error (mCE): 1.000
supervised-only-3d-point-cloud-classificationDGCNN
GFLOPs: 2.4
Number of params (M): 1.8
Overall Accuracy (PB_T50_RS): 78.1

用 AI 构建 AI

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

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

Hyper Newsletters

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