
EMD的Matlab与PyTorch代码分享:适用于3D点云回归的Earth-Mover-Distance (EMD) PyTorch版本...
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
这段简介可以描述为:“EMD的Matlab与PyTorch代码分享”提供了一种新颖的方法,用于处理3D点云数据的回归问题。其中,特别介绍了一个基于Earth-Movers Distance (EMD) 的PyTorch实现版本,旨在简化开发流程并提高计算效率,以促进机器学习社区对3D计算机视觉的研究和应用。
EMD的MATLAB代码用于计算点云之间的地球移动距离(Earth Movers Distance, EMD)已经包装成PyTorch版本。
依赖项:
该代码已在Ubuntu 16.04、PyTorch 1.1.0和CUDA 9.0上进行测试。
使用方法:
首先,通过运行`python setup.py install`来编译。
接着将生成的库文件复制到主目录中。例如:`cp build/lib.linux-x86_64-3.6/emd_cuda.cpython-36m-x86_64-linux-gnu.so .`
之后可以使用以下方式调用:
```python
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包装器是由莫开春编写的,并且顾家远也提供了帮助。
许可协议:MIT许可证。
全部评论 (0)
还没有任何评论哟~


