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!