README.md

Setup

For setting up the requirements to run the code, use the following commands:

# setup train environment
conda env create -f train_environment.yml
conda activate hnf-train
pip install -r train_requirements.txt
pip install git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch
pip install --upgrade pip
pip install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
conda install -c anaconda ipykernel
python -m ipykernel install --user --name=hnf-train
conda deactivate
echo "hnf-train is ready to use! activate with: conda activate hnf-train"

# setup render environment
conda env create -f render_environment.yml
conda activate hnf-render
pip install -r render_requirements.txt
pip install git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch
pip install --upgrade pip
pip install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
conda install -c anaconda ipykernel
python -m ipykernel install --user --name=hnf-render
conda deactivate
echo "hnf-render is ready to use! activate with: conda activate hnf-render"

Downloading checkpoints

For the rendering demo, download the checkpoints from here using:

pip install --upgrade gdown
python3 download_checkpoints.py