inital commit
This commit is contained in:
30
requirements.txt
Normal file
30
requirements.txt
Normal 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
|
||||
Reference in New Issue
Block a user