MogensR commited on
Commit
a226dc7
·
1 Parent(s): 4a91cab

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +46 -100
requirements.txt CHANGED
@@ -1,110 +1,56 @@
1
- # ============================================
2
- # FIXED REQUIREMENTS - MatAnyone compatible
3
- # ============================================
4
-
5
- # Core Dependencies
6
- streamlit==1.32.2
7
- opencv-python-headless==4.8.1.78
8
- opencv-contrib-python==4.8.1.78
9
- numpy==1.26.4
10
- Pillow==10.2.0
11
-
12
- # System Monitoring
13
- psutil==5.9.8
14
- GPUtil==1.4.0
15
- nvidia-ml-py==12.535.161
16
- py3nvml==0.2.7
17
-
18
- # NOTE: PyTorch handled in Dockerfile
19
- # Don't include torch/torchvision/torchaudio here
20
-
21
- # ============================================
22
- # PRIMARY: MatAnyone (with its dependencies)
23
- # ============================================
24
- git+https://github.com/pq-yang/MatAnyone.git
25
-
26
- # ============================================
27
- # SECONDARY: SAM2
28
- # ============================================
29
  git+https://github.com/facebookresearch/segment-anything-2.git
30
 
31
- # ============================================
32
- # TERTIARY: Rembg
33
- # ============================================
34
- rembg==2.0.53
35
- onnxruntime-gpu==1.16.3
36
-
37
- # ============================================
38
- # Supporting Libraries - NO VERSION PINNING!
39
- # Let pip resolve versions automatically
40
- # ============================================
41
-
42
- # Transformers & HuggingFace
43
- transformers
44
- accelerate
45
- huggingface-hub
46
- safetensors
47
- tokenizers
48
-
49
- # Computer Vision
50
- timm
51
- albumentations
52
- ultralytics
53
- supervision
54
-
55
- # Image Processing - NO VERSION for imageio!
56
- scipy
57
- scikit-image
58
- matplotlib
59
- imageio # NO VERSION - let MatAnyone decide!
60
- imageio-ffmpeg
61
-
62
- # Video Processing
63
- moviepy
64
- av
65
-
66
- # ML Libraries
67
- xgboost
68
- lightgbm
69
- scikit-learn
70
-
71
- # Data Processing
72
- pandas
73
- pyarrow
74
 
75
- # Progress & UI
76
- tqdm
77
- rich
78
 
79
- # Web & Downloads
80
- requests
81
- httpx
82
- aiohttp
83
 
84
- # Memory Management
85
- memory-profiler
86
- pympler
87
- objgraph
88
 
89
- # Parallel Processing
90
- joblib
91
- threadpoolctl
92
 
93
- # Utilities
94
- python-dotenv
95
- fire
96
- typer
97
 
98
- # Audio support
99
- librosa
100
- soundfile
101
- pydub
102
 
103
- # Advanced CV models
104
- open-clip-torch
105
- clip-interrogator
106
- groundingdino-py
107
 
108
- # Monitoring
109
- wandb
110
- tensorboard
 
 
 
1
+ # ============================================================================
2
+ # BackgroundFX Pro - Complete Requirements for Hugging Face Spaces
3
+ # ============================================================================
4
+
5
+ # PRIORITY 1: Core Framework
6
+ gradio>=4.0.0,<5.0.0
7
+ torch>=2.0.0,<2.4.0
8
+ torchvision>=0.15.0,<0.19.0
9
+
10
+ # PRIORITY 2: Computer Vision & Image Processing
11
+ opencv-python-headless>=4.8.0,<5.0.0
12
+ pillow>=9.5.0,<11.0.0
13
+ numpy>=1.21.0,<2.0.0
14
+
15
+ # PRIORITY 3: SAM2 Dependencies (Meta's Segment Anything 2)
16
+ # Main SAM2 installation
 
 
 
 
 
 
 
 
 
 
 
 
17
  git+https://github.com/facebookresearch/segment-anything-2.git
18
 
19
+ # SAM2 core dependencies that might not auto-install
20
+ hydra-core>=1.2.0
21
+ iopath>=0.1.9
22
+ fvcore>=0.1.5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
23
 
24
+ # PRIORITY 4: Network & File Handling
25
+ requests>=2.25.0,<3.0.0
26
+ urllib3>=1.26.0,<3.0.0
27
 
28
+ # PRIORITY 5: Scientific Computing & Math
29
+ scipy>=1.7.0,<2.0.0
30
+ scikit-image>=0.19.0,<1.0.0
 
31
 
32
+ # PRIORITY 6: System & Memory Management
33
+ psutil>=5.8.0,<6.0.0
34
+ tqdm>=4.62.0,<5.0.0
 
35
 
36
+ # PRIORITY 7: Data Handling
37
+ packaging>=21.0
38
+ setuptools>=65.0.0
39
 
40
+ # PRIORITY 8: Video Processing Enhancement (Optional but recommended)
41
+ imageio>=2.19.0,<3.0.0
42
+ imageio-ffmpeg>=0.4.7
 
43
 
44
+ # PRIORITY 9: Additional ML/CV utilities
45
+ timm>=0.6.0,<1.0.0
 
 
46
 
47
+ # ============================================================================
48
+ # FALLBACK: If git+https SAM2 installation fails, uncomment below:
49
+ # ============================================================================
50
+ # segment-anything-2
51
 
52
+ # ============================================================================
53
+ # DEVELOPMENT/DEBUG (Remove for production)
54
+ # ============================================================================
55
+ # ipython>=7.0.0
56
+ # jupyter>=1.0.0