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
包装器是由莫开春编写的。此外,顾家远也提供了帮助。
执照
麻省理工学院
2021-09-08 09:33:51
6KB
系统开源
1