34 lines
828 B
Plaintext
34 lines
828 B
Plaintext
# 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
|