EMD的matlab代码分享MSN:用于密集点云完成的变形和采样网络
MSN是一种基于学习的形状补全方法,可以保留已知结构并生成密集且分布均匀的点云。
有关更多详细信息,请参阅我们的
AAAI
2020。
在这个项目中,我们还提供了点云的地球移动距离(EMD)的实现,它基于拍卖算法,只需要
$O(n)$
内存。
完成后获得
32,768
分
用法
1)
环境和先决条件
pytorch
1.2.0
CUDA
10.0
Python
3.7
2)
编译
编译我们的扩展模块:
cd
emd
python3
setup.py
install
cd
expansion_penalty
python3
setup.py
install
cd
MDS
python3
setup.py
install
3)
下载数据和训练好的模型
从
下载数据和训练模型。
由于规模较大,我们不提供训练集的部分点云。
如果要训练模型,可以使用
和
生成它们。
我们为每个
CAD
模型生成
50
个局部点云。
4)
训练或验证
运行python3
val.py来验证模型或python3
train.py从头训练模型。
E
2023-02-24 12:37:19
5.01MB
系统开源
1