MogensR commited on
Commit
4a9bcff
·
1 Parent(s): 0b26cc1

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +33 -29
requirements.txt CHANGED
@@ -1,10 +1,10 @@
1
  # ============================================================================
2
- # CORE PYTHON DEPENDENCIES
3
  # ============================================================================
4
- numpy>=1.24.3,<1.26
5
- Pillow>=10.0.1,<11.0
6
- setuptools>=65.7.0,<69.0
7
- wheel>=0.40.0,<1.0
8
  typing-extensions>=4.12.2,<5.0
9
 
10
  # ============================================================================
@@ -14,53 +14,52 @@ gradio==4.44.1
14
  gradio_client==1.3.0
15
 
16
  # ============================================================================
17
- # DEEP LEARNING & AI MODELS
18
  # ============================================================================
19
- # PyTorch ecosystem (CUDA 12.1 compatible)
20
- torch>=2.0.1,<2.4
21
- torchvision>=0.15.2,<0.19
22
- torchaudio>=2.0.1,<2.4
23
 
24
  # Hugging Face ecosystem
25
- transformers>=4.30.2,<5.0
26
  huggingface_hub>=0.19.3,<1.0
27
- accelerate>=0.20.3,<1.0
28
- safetensors>=0.3.3,<1.0
29
 
30
  # Model utilities
31
- einops>=0.7.0,<1.0
32
- timm>=0.9.5,<1.0
33
 
34
  # ============================================================================
35
  # COMPUTER VISION & VIDEO PROCESSING
36
  # ============================================================================
37
  opencv-python-headless==4.8.0.76
38
 
39
- # Video processing
40
- moviepy>=1.0.3,<2.0
41
- imageio>=2.25.0,<3.0
42
- imageio-ffmpeg>=0.4.8,<1.0
43
- ffmpeg-python>=0.2.0,<1.0
44
- av>=0.5.2,<1.0
45
 
46
  # ============================================================================
47
  # SCIENTIFIC COMPUTING
48
  # ============================================================================
49
- scipy>=1.10.1,<2.0
50
- tqdm>=4.66.1,<5.0
51
 
52
  # ============================================================================
53
  # CONFIGURATION & UTILITIES
54
  # ============================================================================
55
- hydra-core>=1.3.2,<2.0
56
- omegaconf>=2.3.0,<3.0
57
- diskcache>=5.6.3,<6.0
58
  psutil>=5.9.0,<6.0
59
 
60
  # ============================================================================
61
- # MATANYONE DEPENDENCIES
62
  # ============================================================================
63
- # Core dependencies for MatAnyone
64
  cchardet>=2.1.7
65
  cython
66
  easydict
@@ -71,13 +70,18 @@ netifaces>=0.11.0
71
  pycocotools>=2.0.7
72
  tensorboard>=2.11
73
 
 
 
 
 
 
74
  # ============================================================================
75
  # GIT DEPENDENCIES (Model Repositories)
76
  # ============================================================================
77
  # SAM2 - Segment Anything 2
78
  git+https://github.com/facebookresearch/segment-anything-2.git@2b90b9f5ceec907a1c18123530e92e794ad901a4
79
 
80
- # MatAnyone - High-quality matting
81
  git+https://github.com/pq-yang/MatAnyone.git@2234ce5cdc487749515518bd035b5e18bccea3da
82
 
83
  # Thin Plate Spline (MatAnyone dependency)
 
1
  # ============================================================================
2
+ # CORE PYTHON DEPENDENCIES (Python 3.9 compatible)
3
  # ============================================================================
4
+ numpy==1.24.3
5
+ Pillow==10.0.1
6
+ setuptools==65.7.0
7
+ wheel==0.40.0
8
  typing-extensions>=4.12.2,<5.0
9
 
10
  # ============================================================================
 
14
  gradio_client==1.3.0
15
 
16
  # ============================================================================
17
+ # DEEP LEARNING & AI MODELS (Python 3.9 compatible versions)
18
  # ============================================================================
19
+ # PyTorch ecosystem
20
+ torch==2.0.1
21
+ torchvision==0.15.2
22
+ torchaudio==2.0.1
23
 
24
  # Hugging Face ecosystem
25
+ transformers==4.30.2
26
  huggingface_hub>=0.19.3,<1.0
27
+ accelerate==0.20.3
28
+ safetensors==0.3.3
29
 
30
  # Model utilities
31
+ einops==0.7.0
32
+ timm==0.9.5
33
 
34
  # ============================================================================
35
  # COMPUTER VISION & VIDEO PROCESSING
36
  # ============================================================================
37
  opencv-python-headless==4.8.0.76
38
 
39
+ # Video processing (full version with av for MatAnyone)
40
+ moviepy==1.0.3
41
+ imageio==2.25.0
42
+ imageio-ffmpeg==0.4.8
43
+ ffmpeg-python==0.2.0
44
+ av==10.0.0
45
 
46
  # ============================================================================
47
  # SCIENTIFIC COMPUTING
48
  # ============================================================================
49
+ scipy==1.10.1
50
+ tqdm==4.66.1
51
 
52
  # ============================================================================
53
  # CONFIGURATION & UTILITIES
54
  # ============================================================================
55
+ hydra-core==1.3.2
56
+ omegaconf==2.3.0
57
+ diskcache==5.6.3
58
  psutil>=5.9.0,<6.0
59
 
60
  # ============================================================================
61
+ # MATANYONE DEPENDENCIES (Full set for Python 3.9)
62
  # ============================================================================
 
63
  cchardet>=2.1.7
64
  cython
65
  easydict
 
70
  pycocotools>=2.0.7
71
  tensorboard>=2.11
72
 
73
+ # Additional MatAnyone dependencies
74
+ kornia>=0.6.0
75
+ albumentations>=1.3.0
76
+ packaging
77
+
78
  # ============================================================================
79
  # GIT DEPENDENCIES (Model Repositories)
80
  # ============================================================================
81
  # SAM2 - Segment Anything 2
82
  git+https://github.com/facebookresearch/segment-anything-2.git@2b90b9f5ceec907a1c18123530e92e794ad901a4
83
 
84
+ # MatAnyone - High-quality matting (full installation)
85
  git+https://github.com/pq-yang/MatAnyone.git@2234ce5cdc487749515518bd035b5e18bccea3da
86
 
87
  # Thin Plate Spline (MatAnyone dependency)