requirements.txt

--extra-index-url https://download.pytorch.org/whl/nightly/cu124
torch==2.7.0.dev20250218+cu124
torchvision==0.22.0.dev20250219+cu124
roma
gradio==5.17.0
matplotlib
tqdm
opencv-python
scipy
einops
trimesh
tensorboard
git+https://github.com/jensenz-sai/pycolmap@543266bc316df2fe407b3a33d454b310b1641042
pyglet<2
huggingface-hub[torch]>=0.22
pillow-heif  # add heif/heic image support
pyrender  # for rendering depths in scannetpp
kapture  # for visloc data loading
kapture-localization
numpy==1.24.4
numpy-quaternion
pycolmap  # for pnp
poselib  # for pnp
viser
tyro
ninja
colorama
pytorch-lightning
splines
diffusers
kornia
open-clip-torch
accelerate
pyav
imageio[ffmpeg]