# Core deep learning and computer vision
torch>=2.0.0
torchvision>=0.15.0
ultralytics>=8.0.0
opencv-python>=4.8.0
numpy>=1.24.0

# SAM2 - Segment Anything Model 2
# Note: Make sure to run download_models.py after installing to get model weights
git+https://github.com/facebookresearch/sam2.git

# GPU acceleration (optional but recommended)
cupy-cuda12x>=12.0.0  # For CUDA 12.x, adjust version as needed

# Configuration and utilities
PyYAML>=6.0
tqdm>=4.65.0
matplotlib>=3.7.0
Pillow>=10.0.0

decord

# Optional: For advanced features
psutil>=5.9.0          # Memory monitoring
pympler>=0.9           # Memory profiling (for debugging)

# Video processing
ffmpeg-python>=0.2.0   # Python wrapper for FFmpeg (optional, shell ffmpeg still needed)

# Development dependencies (optional)
pytest>=7.0.0
black>=23.0.0
flake8>=6.0.0