Skip to content
No description, website, or topics provided.
Python
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
pretrained first commit May 8, 2019
data.py first commit May 8, 2019
main.py first commit May 8, 2019
model.py first commit May 8, 2019
readme.md Update readme.md Jun 15, 2019
util.py

readme.md

Deep Cloeset Point

Prerequisites

PyTorch>=1.0: https://pytorch.org

scipy>=1.2

numpy

h5py

tqdm

TensorboardX: https://github.com/lanpa/tensorboardX

Training

DCP-v1

python main.py --exp_name=dcp_v1 --model=dcp --emb_nn=dgcnn --pointer=identity --head=svd

DCP-v2

python main.py --exp_name=dcp_v2 --model=dcp --emb_nn=dgcnn --pointer=transformer --head=svd

Testing

DCP-v1

python main.py --exp_name=dcp_v1 --model=dcp --emb_nn=dgcnn --pointer=identity --head=svd --eval

or

python main.py --exp_name=dcp_v1 --model=dcp --emb_nn=dgcnn --pointer=identity --head=svd --eval --model_path=xx/yy

DCP-v2

python main.py --exp_name=dcp_v2 --model=dcp --emb_nn=dgcnn --pointer=transformer --head=svd --eval

or

python main.py --exp_name=dcp_v2 --model=dcp --emb_nn=dgcnn --pointer=transformer --head=svd --eval --model_path=xx/yy

where xx/yy is the pretrained model

Citation

Please cite this paper if you want to use it in your work,

@article{dcp,
  title={Deep Closest Point: Learning Representations for Point Cloud Registration},
  author={Wang, Yue and Solomon, Justin M.},
  journal={arXiv preprint arXiv:1905.03304},
  year={2019}
}

License

MIT License

You can’t perform that action at this time.