-
Notifications
You must be signed in to change notification settings - Fork 17
Expand file tree
/
Copy pathrequirements.docker.txt
More file actions
88 lines (77 loc) · 1.97 KB
/
requirements.docker.txt
File metadata and controls
88 lines (77 loc) · 1.97 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
# =================================================================================
# GENERAL DEPENDENCIES
# Contains all other project requirements (core libs, add-ons, annotators.
# =================================================================================
#__________________________________________________________________________________
# 1) Core Python packages
numpy==1.26.3
matplotlib==3.9.2
pillow==11.0.0
tqdm==4.67.1
opencv-python-headless==4.9.0.80
PyYAML==6.0.2
einops==0.8.1
huggingface-hub[cli]==0.29.2
imageio==2.36.0
timm==1.0.19
safetensors==0.6.2
scipy==1.15.3
wandb==0.21.3
gdown
#__________________________________________________________________________________
# 2) Related deps (torch-scatter is separate)
ninja==1.11.1
kornia==0.7.4
plotly
#__________________________________________________________________________________
# 3) Gsam annotator
gradio
supervision
hydra-core
transformers==4.49.0
iopath
#__________________________________________________________________________________
# 4) Scene Inpainting
diffusers==0.30.0
#__________________________________________________________________________________
# 5) Scene Points Generation
open3d
git+https://github.com/microsoft/MoGe.git@72fdee98935d93b69f7ce0a1369cfd6e2573fc2f
#__________________________________________________________________________________
# 6) Object mesh generation
rembg
onnxruntime-gpu==1.17.0
pymeshlab==2023.12.post3
xatlas
pybind11
pygltflib
accelerate
#__________________________________________________________________________________
# 7) Object scene alignment
probreg
Rtree
# 8) Foundation Pose
transformations==2024.6.1
ruamel.yaml==0.18.6
h5py==3.10.0
# 9) PromptDA
pyliblzfse
lightning==2.1.3
termcolor
tyro==0.9.2
spaces
trimesh
# 10) WiLoR
pyrender
pytorch-lightning
scikit-image
smplx==0.1.28
yacs
xtcocotools
pandas
hydra-submitit-launcher
hydra-colorlog
pyrootutils
webdataset
ultralytics==8.1.34
# Note: chumpy installed separately in Dockerfile due to build isolation issues