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 1

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

11
machine-learning/immich_ml/schemas.py
View file

@ -23,14 +23,14 @@ class BoundingBox(TypedDict):
class ModelTask(StrEnum):
FACIAL_RECOGNITION = "facial-recognition"
SEARCH = "clip"
OCR = "ocr"
class ModelType(StrEnum):
DETECTION = "detection"
RECOGNITION = "recognition"
TEXTUAL = "textual"
VISUAL = "visual"
OCR = "ocr"
class ModelFormat(StrEnum):
ARMNN = "armnn"
@ -42,7 +42,7 @@ class ModelSource(StrEnum):
INSIGHTFACE = "insightface"
MCLIP = "mclip"
OPENCLIP = "openclip"
PADDLE = "paddle"
ModelIdentity = tuple[ModelType, ModelTask]
@ -87,6 +87,11 @@ class DetectedFace(TypedDict):
FacialRecognitionOutput = list[DetectedFace]
class OCROutput(TypedDict):
text: str
confidence: float
class PipelineEntry(TypedDict):
modelName: str
options: dict[str, Any]