requirements.txt

easydict
einops
fastapi
gdown
gradio<=5.4.0
h5py
huggingface_hub
imageio
Jinja2
kornia
loguru
matplotlib<3.9
numpy~=1.24
omegaconf
onnxruntime
opencv-contrib-python
opencv-python
pandas
plotly
poselib
protobuf
psutil
pycolmap==0.6.1
pytlsd
pytorch-lightning==1.4.9
PyYAML
scikit-image
scikit-learn
scipy
seaborn
shapely
tensorboardX==2.6.1
torch==2.0.0
torchvision==0.15.1
torchmetrics==0.6.0
tqdm
uvicorn
yacs
safetensors
pynvml