EMD的matlab代码分享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从头训练模型。