You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Pang-Yatian
336d42f562
|
2 years ago | |
|---|---|---|
| .. | ||
| cuda | 2 years ago | |
| .gitignore | 2 years ago | |
| README.md | 2 years ago | |
| __init__.py | 2 years ago | |
| emd.py | 2 years ago | |
| setup.py | 2 years ago | |
| test_emd_loss.py | 2 years ago | |
README.md
PyTorch Wrapper for Point-cloud Earth-Mover-Distance (EMD)
Dependency
The code has been tested on Ubuntu 16.04, PyTorch 1.1.0, CUDA 9.0.
Usage
First compile using
python setup.py install
Then, copy the lib file out to the main directory,
cp build/lib.linux-x86_64-3.6/emd_cuda.cpython-36m-x86_64-linux-gnu.so .
Then, you can use it by simply
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
Check test_emd_loss.py for example.
Author
The cuda code is originally written by Haoqiang Fan. The PyTorch wrapper is written by Kaichun Mo. Also, Jiayuan Gu provided helps.
License
MIT
No Description
Python Cuda Markdown C++ other