KDnet
This repository is an independent implementation of https://arxiv.org/abs/1704.01222 in pytorch.
Data download and training
download bash download.sh
, training: python train.py
Dependencies
- scipy
- pytorch==0.3.1 (not tested in other versions)
Acknowledgements
Shout out to @wassname who increased the scalibility upon my original KDTree implementation!