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 7

fix(ocr): switch RapidOCR lang type from LangDet to LangRec

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

4
machine-learning/immich_ml/models/ocr/recognition.py
View file

@ -7,7 +7,7 @@ from rapidocr.ch_ppocr_rec import TextRecInput
from rapidocr.ch_ppocr_rec import TextRecognizer as RapidTextRecognizer from rapidocr.ch_ppocr_rec import TextRecognizer as RapidTextRecognizer
from rapidocr.inference_engine.base import FileInfo, InferSession from rapidocr.inference_engine.base import FileInfo, InferSession
from rapidocr.utils import DownloadFile, DownloadFileInput from rapidocr.utils import DownloadFile, DownloadFileInput
from rapidocr.utils.typings import EngineType, LangDet, OCRVersion, TaskType from rapidocr.utils.typings import EngineType, LangRec, OCRVersion, TaskType
from rapidocr.utils.typings import ModelType as RapidModelType from rapidocr.utils.typings import ModelType as RapidModelType
from immich_ml.config import log, settings from immich_ml.config import log, settings
@ -38,7 +38,7 @@ class TextRecognizer(InferenceModel):
engine_type=EngineType.ONNXRUNTIME, engine_type=EngineType.ONNXRUNTIME,
ocr_version=OCRVersion.PPOCRV5, ocr_version=OCRVersion.PPOCRV5,
task_type=TaskType.REC, task_type=TaskType.REC,
lang_type=LangDet.CH, lang_type=LangRec.CH,
model_type=RapidModelType.MOBILE if "mobile" in self.model_name else RapidModelType.SERVER, model_type=RapidModelType.MOBILE if "mobile" in self.model_name else RapidModelType.SERVER,
) )
) )