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

fix(ocr): expose lang_type (LangRec.CH) and font_path on OcrOptions for RapidOCR

Browse source
This commit is contained in:
CoderKang 2025-09-09 23:14:35 +08:00
parent 7211086cb2
commit b5a2bf5ae5
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
machine-learning/immich_ml/models/ocr/schemas.py
View file

@ -2,7 +2,7 @@ from typing import Iterable
import numpy as np
import numpy.typing as npt
from rapidocr.utils.typings import EngineType
from rapidocr.utils.typings import EngineType, LangRec
from typing_extensions import TypedDict
@ -19,8 +19,10 @@ class TextRecognitionOutput(TypedDict):
textScore: npt.NDArray[np.float32]
# RapidOCR expects engine_type to be an attribute
# RapidOCR expects `engine_type`, `lang_type`, and `font_path` to be attributes
class OcrOptions(dict):
def __init__(self, **options):
super().__init__(**options)
self.engine_type = EngineType.ONNXRUNTIME
self.lang_type = LangRec.CH
self.font_path = None