helpy4/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich synced 2025-11-07 17:27:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: add OCR functionality and related configurations

Browse source
This commit is contained in:
CoderKang 2025-06-01 22:10:43 +08:00 • committed by mertalev
parent 23fb2e0fae
commit 0e8ca1c159
No known key found for this signature in database
GPG key ID: DF6ABC77AAD98C95
64 changed files with 3998 additions and 1669 deletions
Download patch file Download diff file Expand all files Collapse all files

5
machine-learning/immich_ml/models/__init__.py
View file

@ -8,7 +8,7 @@ from immich_ml.schemas import ModelSource, ModelTask, ModelType
from .constants import get_model_source
from .facial_recognition.detection import FaceDetector
from .facial_recognition.recognition import FaceRecognizer
from .ocr.paddle import PaddleOCRecognizer
def get_model_class(model_name: str, model_type: ModelType, model_task: ModelTask) -> type[InferenceModel]:
source = get_model_source(model_name)
@ -28,6 +28,9 @@ def get_model_class(model_name: str, model_type: ModelType, model_task: ModelTas
case ModelSource.INSIGHTFACE, ModelType.RECOGNITION, ModelTask.FACIAL_RECOGNITION:
return FaceRecognizer
case ModelSource.PADDLE, ModelType.OCR, ModelTask.OCR:
return PaddleOCRecognizer
case _:
raise ValueError(f"Unknown model combination: {source}, {model_type}, {model_task}")