requirements.txt

# ─────────────────────────────────────────────────────────────
# Car Segmentation - 4 Model Comparison
# Requirements File
# ─────────────────────────────────────────────────────────────
# Install command:
#   pip install -r requirements.txt
#
# GPU (CUDA) users ke liye PyTorch alag install karna hoga:
#   pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118
# CPU users seedha install kar saktay hain (requirements mein already hai)
# ─────────────────────────────────────────────────────────────

# ── Core ──────────────────────────────────────────────────────
numpy>=1.24.0
opencv-python>=4.8.0
Pillow>=9.5.0
matplotlib>=3.7.0

# ── PyTorch (CPU version) ──────────────────────────────────────
# GPU hai toh neeche wali lines comment karke upar wala GPU command chalao
torch>=2.0.0
torchvision>=0.15.0

# ── Model 1: YOLOv8-seg ───────────────────────────────────────
ultralytics>=8.0.0

# ── Model 2: Mask R-CNN ───────────────────────────────────────
# torchvision already upar install ho gayi (MaskRCNN uski andar hai)

# ── Model 3: SegFormer ────────────────────────────────────────
transformers>=4.35.0
accelerate>=0.24.0

# ── Model 4: SAM (Segment Anything Model) ─────────────────────
segment-anything>=1.0

# ── Utilities ─────────────────────────────────────────────────
requests>=2.28.0          # weights download ke liye
tqdm>=4.65.0              # progress bars
gradio>=3.40.0            # Web UI for Hugging Face Spaces

transformers

einops
kornia
timm