Enhanced-Zhang-Colorization-with-Object-Aware-Processing/requirements.txt at main · Hadifard/Enhanced-Zhang-Colorization-with-Object-Aware-Processing · GitHub

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
# Zhang Colorization Enhanced - Requirements
# Python 3.7+ required

# ============================================================================
# Core Dependencies (Required)
# ============================================================================

# Computer Vision
opencv-python>=4.5.0,<5.0.0
opencv-contrib-python>=4.5.0,<5.0.0

# Numerical Computing
numpy>=1.19.0,<2.0.0

# Deep Learning Framework
torch>=1.9.0,<3.0.0
torchvision>=0.10.0,<1.0.0

# Image Processing
Pillow>=8.0.0,<11.0.0

# ============================================================================
# Optional Dependencies
# ============================================================================

# Facial Landmark Detection (for facial_feature_enhancement.py)
# Uncomment if you need facial feature enhancement
# dlib>=19.22.0

# YOLOv8 Segmentation (for better object detection)
# Uncomment if you want superior segmentation quality
# ultralytics>=8.0.0

# ============================================================================
# Utility Libraries
# ============================================================================

# Path handling (usually built-in, but ensure compatibility)
pathlib2>=2.3.0; python_version < '3.4'

# Type hints support
typing-extensions>=4.0.0; python_version < '3.8'

# ============================================================================
# Installation Notes
# ============================================================================
#
# Basic Installation (Minimum requirements):
#   pip install -r requirements.txt
#
# Full Installation (With all optional features):
#   pip install -r requirements.txt
#   pip install dlib ultralytics
#
# For CUDA Support (GPU acceleration):
#   pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118
#   (Replace cu118 with your CUDA version: cu117, cu118, cu121, etc.)
#
# For Apple Silicon (M1/M2/M3):
#   pip install torch torchvision
#   (PyTorch has native Metal Performance Shaders support)
#
# Troubleshooting dlib installation:
#   - Ubuntu/Debian: sudo apt-get install cmake libboost-all-dev
#   - macOS: brew install cmake boost
#   - Windows: Download pre-built wheel from https://github.com/sachadee/Dlib
#
# ============================================================================