requirements.txt

numpy==1.21.5
scipy==1.7.3
imageio==2.19.3
opencv-python==4.5.5.62
openmesh==1.1.6
trimesh==3.9.42
Pillow==9.0
psutil==5.9.0
tqdm==4.62.3
matplotlib==3.5.2
lpips==0.1.4
pytorch-msssim==1.0.0
# tsdf-fusion-python
numba==0.55.0
pycuda==2021.1
scikit-image==0.19.1
# trimesh icp
rtree==0.9.7
plyfile==0.7.4
# DPT Depth Estimation
timm==0.5.4
# Consistent Video Depth Estimation
wget==3.2
gdown==4.6.0
h5py==3.6.0
pypng==0.0.21
# Creating Excel spreadsheets with Pandas
openpyxl==3.0.10
# Packages from the LaMa integration branch
debugpy==1.6.7
gradio==3.39.0
docker==6.1.3
python-on-whales==0.63.0
pyyaml
easydict==1.9.0
scikit-learn==0.24.2
tensorflow==2.13.0
joblib==1.3.1
albumentations==0.5.2
hydra-core==1.1.0
pytorch-lightning==1.2.9
tabulate==0.9.0
kornia==0.5.0
webdataset==0.2.48
packaging==23.1
scikit-learn==0.24.2
# For headless rendering with trimesh
pyglet==1.5.27
pyrender==0.1.45
# Needs to be this version in order to run the FlowNet2 install.sh script.
# Later versions will error on CUDA version mismatch instead of warn.
setuptools==59.5.0
networkx==3.1  # Can't use Networkx 3.2.1, the latest version from the packages list below, since it requires python 3.9
-f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu113_pyt1100/download.html
pytorch3d
# General
-f https://download.pytorch.org/whl/cu113/torch_stable.html
torch==1.10.2+cu113
torchvision==0.11.3+cu113
-f https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html
detectron2