MogensR commited on
Commit
5e08b76
·
verified ·
1 Parent(s): 30c74de

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +47 -45
requirements.txt CHANGED
@@ -1,76 +1,78 @@
1
  # ===============================
2
  # BackgroundFX Pro — Requirements
3
  # Python 3.10 + CUDA 12.1 + Torch 2.5.1
4
- # SAM2 + MatAnyone Integration
5
  # ===============================
6
 
7
  # --- PyTorch (Must be first) ---
8
- torch>=2.5.1,<2.6.0
9
- torchvision>=0.20.1,<0.21.0
10
 
11
  # --- Core ---
12
- numpy>=1.24.0,<2.0.0
13
- Pillow>=10.0.0,<11.0.0
14
- protobuf>=4.25.0,<5.0.0
15
 
16
  # --- Image / Video Processing ---
17
- opencv-python-headless>=4.8.0,<5.0.0
18
- imageio>=2.25.0,<3.0.0
19
- imageio-ffmpeg>=0.4.7,<1.0.0
20
- moviepy>=1.0.3,<2.0.0
21
- decord>=0.6.0,<0.7.0
22
- scikit-image>=0.19.3,<0.22.0
23
 
24
  # --- MediaPipe ---
25
- mediapipe>=0.10.0,<0.11.0
26
 
27
  # --- SAM2 Dependencies ---
28
- hydra-core>=1.3.2,<2.0.0
29
- omegaconf>=2.3.0,<3.0.0
30
- einops>=0.6.0,<0.9.0
31
- timm>=0.9.0,<1.1.0
32
- pyyaml>=6.0.0,<7.0.0
33
- matplotlib>=3.5.0,<4.0.0
34
- iopath>=0.1.10,<0.2.0
35
 
36
  # --- SAM2 Model (GitHub installation) ---
37
  git+https://github.com/facebookresearch/segment-anything-2.git
38
 
39
  # --- MatAnyone Dependencies ---
40
- kornia>=0.7.2,<0.8.0 # 0.7.2 tested with Torch 2.5.x
41
- tqdm>=4.60.0,<5.0.0
42
- scipy>=1.10.1,<2.0.0 # Required for MatAnyone
43
 
44
  # --- MatAnyone Model (GitHub installation) ---
45
  git+https://github.com/pq-yang/MatAnyone.git
46
 
47
  # --- Streamlit UI ---
48
- streamlit>=1.32.0,<2.0.0
49
- streamlit-webrtc>=0.50.0,<1.0.0
50
 
51
- # --- Web stack ---
52
- fastapi>=0.104.0,<1.0.0
53
- uvicorn>=0.24.0,<1.0.0
54
- httpx>=0.25.0,<1.0.0
55
- anyio>=4.0.0,<5.0.0
56
- orjson>=3.10.0,<4.0.0
57
 
58
  # --- Pydantic / Typing ---
59
- pydantic>=2.6.0,<2.9.0
60
- pydantic-core>=2.20.0,<2.22.0
61
- annotated-types>=0.6.0,<0.8.0
62
- typing-extensions>=4.0.0,<5.0.0
63
 
64
  # --- Helpers / Utilities ---
65
- huggingface-hub>=0.20.0,<1.0.0
66
- ffmpeg-python>=0.2.0,<1.0.0
67
- psutil>=5.8.0,<7.0.0
68
- requests>=2.25.0,<3.0.0
69
- scikit-learn>=1.3.0,<2.0.0
70
- loguru>=0.6.0,<1.0.0
71
- python-multipart>=0.0.5,<1.0.0
72
 
73
  # --- Additional for SAM2 + MatAnyone ---
74
- transformers>=4.30.0,<5.0.0
75
- accelerate>=0.20.0,<1.0.0
76
- ninja>=1.10.0,<2.0.0 # For faster CUDA compilation
 
 
 
 
 
 
 
1
  # ===============================
2
  # BackgroundFX Pro — Requirements
3
  # Python 3.10 + CUDA 12.1 + Torch 2.5.1
4
+ # SAM2 + MatAnyone Integration with Audio Support
5
  # ===============================
6
 
7
  # --- PyTorch (Must be first) ---
8
+ torch==2.5.1
9
+ torchvision==0.20.1
10
 
11
  # --- Core ---
12
+ numpy==1.24.0
13
+ Pillow==10.0.0
14
+ protobuf==4.25.0
15
 
16
  # --- Image / Video Processing ---
17
+ opencv-python-headless==4.8.0
18
+ imageio==2.25.0
19
+ imageio-ffmpeg==0.4.7
20
+ moviepy==1.0.3
21
+ decord==0.6.0
22
+ scikit-image==0.19.3
23
 
24
  # --- MediaPipe ---
25
+ mediapipe==0.10.0
26
 
27
  # --- SAM2 Dependencies ---
28
+ hydra-core==1.3.2
29
+ omegaconf==2.3.0
30
+ einops==0.6.0
31
+ timm==0.9.0
32
+ pyyaml==6.0.0
33
+ matplotlib==3.5.0
34
+ iopath==0.1.10
35
 
36
  # --- SAM2 Model (GitHub installation) ---
37
  git+https://github.com/facebookresearch/segment-anything-2.git
38
 
39
  # --- MatAnyone Dependencies ---
40
+ kornia==0.7.2
41
+ tqdm==4.65.0
42
+ scipy==1.10.1
43
 
44
  # --- MatAnyone Model (GitHub installation) ---
45
  git+https://github.com/pq-yang/MatAnyone.git
46
 
47
  # --- Streamlit UI ---
48
+ streamlit==1.32.0
 
49
 
50
+ # --- Web stack (for potential API extensions) ---
51
+ fastapi==0.104.0
52
+ uvicorn==0.24.0
 
 
 
53
 
54
  # --- Pydantic / Typing ---
55
+ pydantic==2.6.0
56
+ pydantic-core==2.20.0
57
+ annotated-types==0.6.0
58
+ typing-extensions==4.0.0
59
 
60
  # --- Helpers / Utilities ---
61
+ huggingface-hub==0.20.0
62
+ ffmpeg-python==0.2.0
63
+ psutil==5.9.4
64
+ requests==2.25.0
65
+ scikit-learn==1.3.0
66
+ loguru==0.6.0
67
+ python-multipart==0.0.5
68
 
69
  # --- Additional for SAM2 + MatAnyone ---
70
+ transformers==4.30.0
71
+ accelerate==0.20.0
72
+ ninja==1.10.0
73
+
74
+ # --- Logging ---
75
+ logging==0.4.9.6
76
+
77
+ # --- For better file handling ---
78
+ aiofiles==23.1.0