dtoro
9282a5c734
Introduce the app/services/vision/ module with ABC interfaces, ONNX Runtime backend, model registry, and per-task implementations: - OpenCLIP ViT-B/32 embedder (image + text, 512-d) - RapidOCR engine (PP-OCRv4 via ONNX, no PaddlePaddle) - YOLOv8n object detector (raw ONNX, no ultralytics runtime) - YuNet + SFace face processor (Apache 2.0, opencv_zoo, 128-d) - DBSCAN face clustering helper Add VisionSettings to config (mulita.yml + Pydantic), bootstrap_models.py for first-boot weight downloads, models_data Docker volume, and ROCm backend stub for future GPU acceleration. No Celery tasks wired yet — models load but nothing invokes them. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
60 lines
1.4 KiB
Plaintext
60 lines
1.4 KiB
Plaintext
# Core dependencies
|
|
fastapi==0.109.0
|
|
uvicorn[standard]==0.27.0
|
|
python-multipart==0.0.6
|
|
|
|
# Database
|
|
sqlalchemy[asyncio]==2.0.25
|
|
aiosqlite==0.19.0 # SQLite escape hatch (docker-compose.sqlite.yml override)
|
|
asyncpg==0.29.0 # async Postgres driver (default)
|
|
psycopg2-binary==2.9.9 # sync Postgres driver, used by Alembic CLI
|
|
pgvector==0.2.5 # pgvector SQLAlchemy types
|
|
alembic==1.13.1
|
|
|
|
# Redis and Celery
|
|
redis==5.0.1
|
|
celery==5.3.6
|
|
flower==2.0.1
|
|
|
|
# Image processing
|
|
# pyvips==2.2.1 # Optional - having compatibility issues, using Pillow as fallback
|
|
# rawpy==0.19.0 # Optional - numpy compatibility issues, using Pillow as fallback
|
|
pillow==10.2.0
|
|
pillow-heif==0.15.0
|
|
imagehash==4.3.1 # perceptual hash for duplicate detection
|
|
imageio==2.33.1
|
|
imageio-ffmpeg==0.4.9
|
|
|
|
# Video processing
|
|
ffmpeg-python==0.2.0
|
|
|
|
# Metadata extraction
|
|
pyexiftool==0.5.6
|
|
|
|
# File watching
|
|
watchfiles==0.21.0
|
|
|
|
# Vision pipeline (ONNX Runtime CPU inference)
|
|
onnxruntime==1.17.1
|
|
open-clip-torch==2.24.0 # tokenizer + export helper; inference via ONNX
|
|
rapidocr-onnxruntime==1.3.22
|
|
scikit-learn==1.4.0 # DBSCAN for face clustering
|
|
numpy>=1.26.0,<2.0
|
|
|
|
# Utilities
|
|
pyyaml==6.0.1
|
|
pydantic==2.5.3
|
|
pydantic-settings==2.1.0
|
|
python-dotenv==1.0.0
|
|
httpx==0.26.0
|
|
aiofiles==23.2.1
|
|
|
|
# Security and authentication
|
|
python-jose[cryptography]==3.3.0
|
|
passlib[bcrypt]==1.7.4
|
|
|
|
# Development
|
|
pytest==7.4.4
|
|
pytest-asyncio==0.23.3
|
|
black==23.12.1
|
|
ruff==0.1.11 |