MogensR commited on
Commit
4f4dff5
·
1 Parent(s): 86debcc

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +33 -40
requirements.txt CHANGED
@@ -1,10 +1,10 @@
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,74 +14,67 @@ gradio==4.44.1
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
66
  gdown>=4.7.1
67
- gitpython>=3.1
68
  hickle>=5.0
 
 
69
  netifaces>=0.11.0
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)
 
1
  # ============================================================================
2
+ # CORE PYTHON DEPENDENCIES (Python 3.10)
3
  # ============================================================================
4
+ numpy==1.26.4
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
  gradio_client==1.3.0
15
 
16
  # ============================================================================
17
+ # DEEP LEARNING & AI MODELS (Python 3.10 + CUDA 12.1)
18
  # ============================================================================
19
+ # PyTorch ecosystem - CUDA 12.1 compatible
20
+ torch==2.1.0
21
+ torchvision==0.16.0
22
+ torchaudio==2.1.0
23
 
24
+ # Hugging Face ecosystem - MatAnyone compatible versions
25
+ transformers==4.43.3
26
+ huggingface_hub==0.24.5
27
+ accelerate>=0.20.3,<1.0
28
+ safetensors==0.4.3
29
 
30
+ # Model utilities - MatAnyone requirements
31
+ einops==0.8.0
32
+ timm>=0.9.16
33
 
34
  # ============================================================================
35
  # COMPUTER VISION & VIDEO PROCESSING
36
  # ============================================================================
37
+ opencv-python-headless==4.10.0.84
38
 
39
+ # Video processing
40
+ moviepy>=1.0.3,<2.0
41
+ imageio==2.34
42
+ imageio-ffmpeg>=0.4.8,<1.0
43
+ ffmpeg-python>=0.2.0,<1.0
 
44
 
45
  # ============================================================================
46
  # SCIENTIFIC COMPUTING
47
  # ============================================================================
48
+ scipy==1.13.1
49
+ tqdm>=4.66.1,<5.0
50
 
51
  # ============================================================================
52
  # CONFIGURATION & UTILITIES
53
  # ============================================================================
54
  hydra-core==1.3.2
55
  omegaconf==2.3.0
56
+ diskcache>=5.6.3,<6.0
57
  psutil>=5.9.0,<6.0
58
 
59
  # ============================================================================
60
+ # MATANYONE DEPENDENCIES (Python 3.10 compatible)
61
  # ============================================================================
62
+ easydict==0.1.10
 
 
63
  gdown>=4.7.1
 
64
  hickle>=5.0
65
+ cchardet>=2.1.7
66
+ gitpython>=3.1
67
  netifaces>=0.11.0
68
  pycocotools>=2.0.7
69
  tensorboard>=2.11
70
 
 
 
 
 
 
71
  # ============================================================================
72
  # GIT DEPENDENCIES (Model Repositories)
73
  # ============================================================================
74
+ # SAM2 - Latest version (Python 3.10+ compatible)
75
  git+https://github.com/facebookresearch/segment-anything-2.git@2b90b9f5ceec907a1c18123530e92e794ad901a4
76
 
77
+ # MatAnyone - Full installation (Python 3.10 compatible)
78
  git+https://github.com/pq-yang/MatAnyone.git@2234ce5cdc487749515518bd035b5e18bccea3da
79
 
80
  # Thin Plate Spline (MatAnyone dependency)