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

Shout out to @wassname who increased the scalibility upon my original KDTree implementation!

Name		Name	Last commit message	Last commit date
Latest commit History 32 Commits
License		License
README.md		README.md
build.sh		build.sh
dataset_metallic_glass.py		dataset_metallic_glass.py
datasets.py		datasets.py
download.sh		download.sh
kdnet.py		kdnet.py
kdtree.py		kdtree.py
playground.ipynb		playground.ipynb
render_balls_so.cpp		render_balls_so.cpp
show3d_balls.py		show3d_balls.py
test.py		test.py
train.py		train.py
train_MG.py		train_MG.py
train_MG2.py		train_MG2.py
train_batch.py		train_batch.py