4 个月前

PointFlow:基于连续归一化流的3D点云生成

PointFlow:基于连续归一化流的3D点云生成

摘要

随着三维点云成为多种视觉和图形应用的首选表示形式,合成或重建高分辨率、高保真度点云的能力变得至关重要。尽管深度学习模型在点云的判别任务中取得了近期的成功,但生成点云仍然具有挑战性。本文提出了一种基于概率原理的框架,通过将点云建模为分布的分布来生成三维点云。具体而言,我们学习了两个层次的分布:第一层次是形状的分布,第二层次是在给定形状下的点的分布。这种表述使我们既能采样形状,也能从一个形状中采样任意数量的点。我们的生成模型命名为PointFlow(点流),该模型使用连续归一化流来学习每个层次的分布。归一化流的可逆性使得在训练过程中可以计算似然值,并允许我们在变分推断框架下训练模型。实验结果表明,PointFlow在点云生成方面达到了最先进的性能。此外,我们还展示了该模型能够忠实地重建点云,并以无监督的方式学习有用的表示。代码将在https://github.com/stevenygd/PointFlow上提供。

代码仓库

rg321/pointFlow_exp
pytorch
GitHub 中提及
maciejzieba/HyperFlow
pytorch
GitHub 中提及
lmy1001/pointflow_experiment
pytorch
GitHub 中提及
rameenabdal/styleflow
pytorch
GitHub 中提及
gmum/LoCondA
pytorch
GitHub 中提及
theamaya/3dlatnav
pytorch
GitHub 中提及
AnTao97/FoldingNet.pytorch
pytorch
GitHub 中提及
stevenygd/PointFlow
官方
pytorch
GitHub 中提及
justinjohn0306/StyleFlow
tf
GitHub 中提及
visinf/s2-flow
pytorch
GitHub 中提及

基准测试

基准方法指标
point-cloud-generation-on-shapenet-airplanePointFlow
1-NNA-CD: 75.68
MMD-CD: 0.217
point-cloud-generation-on-shapenet-carPointFlow
1-NNA-CD: 60.65
MMD-CD: 0.91
point-cloud-generation-on-shapenet-chairPointFlow
1-NNA-CD: 60.88
MMD-CD: 2.42

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
PointFlow:基于连续归一化流的3D点云生成 | 论文 | HyperAI超神经