- OS: Ubuntu 20.04.2 LTS
- GPU: >= NVIDIA GTX 1080 Ti with >= 460.73.01 driver
- Python package manager
conda
conda env create -f environment.yml
source activate kiloosf-env
please make a copy of cuda in the root directory of this repo and follow the steps in CUDA extension installation.
To train a model yourself run
bash train.sh
To test a trained model on the testing set, run
bash benchmark.sh
To render spiral demo, run
bash demo.sh
You can adjust the dataset by setting the dataset variable in the respective script.
Part of our code is derived from kilenerf, thanks to the authors for the great works.