inital commit

This commit is contained in:
2025-07-27 11:43:07 -07:00
commit b20a8281e5
13 changed files with 1877 additions and 0 deletions

30
requirements.txt Normal file
View File

@@ -0,0 +1,30 @@
# 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
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
# 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