EMD的matlab代码分享-PyTorchEMD:用于3D点云回归的Earth-Mover-Distance(EMD)的PyT...

EMD的matlab代码分享用于点云地球移动距离 (EMD) 的 PyTorch 包装器 依赖 代码已经在 Ubuntu 16.04、PyTorch 1.1.0、CUDA 9.0 上测试过。 用法 首先编译使用 python setup.py install 然后，将lib文件复制到主目录， cp build/lib.linux-x86_64-3.6/emd_cuda.cpython-36m-x86_64-linux-gnu.so . 然后，您可以简单地使用它 from emd import earth_mover_distance d = earth_mover_distance(p1, p2, transpose=False) # p1: B x N1 x 3, p2: B x N2 x 3 例如检查test_emd_loss.py 。 作者 cuda代码最初由范浩强编写。 PyTorch 包装器是由莫开春编写的。此外，顾家远也提供了帮助。 执照 麻省理工学院