Sparse Voxels Rasterization (SVRaster)

Tutorial

# Create a Python environment for SVRaster
conda create -n svraster python=3.9
conda activate svraster
 
# Install Dependencies
pip install torch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/cu124
conda install -y -c "nvidia/label/cuda-12.4.0" cuda-toolkit
 
git clone https://github.com/NVlabs/svraster.git
cd svraster
 
pip install -r requirements.txt
 
cd cuda
pip install -e .
 
cd ..
pip install viser
 
# Setup 
chmod +x run_colmap.sh
 
mkdir data
cd data
 
# create project under data directory
svraster/
├── run_colmap.sh
├── data/
   └── bonsai-1/
       ├── images/              # <-- Place input images here
       └──                      # database.db and colmap results will be generated here
 
# Run Script
./run_colmap.sh bonsai-1 --matcher exhaustive
 
# Train Model
python train.py --eval --source_path data/bonsai-1 --model_path result/bonsai-1 --lambda_T_inside 0.01 --lambda_normal_dmean 0.001 --lambda_normal_dmed 0.001 --lambda_sparse_depth 0.01
 
# View Model
python viz.py result/bonsai-1

Resources