superrob1500
/
mirror-immich
mirror of https://github.com/immich-app/immich
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix(ml): use self.cache_dir for model_root_dir in OcrOptions

pull/28610/head
pneuly 2026-05-27 01:01:02 +00:00
parent 0ef16cf14f
commit 4fd605f81f
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
machine-learning/immich_ml/models/ocr/recognition.py
View File

@ -64,7 +64,7 @@ class TextRecognizer(InferenceModel):
rec_batch_num=max_batch_size if max_batch_size else 6,
rec_img_shape=(3, 48, 320),
lang_type=self.language,
model_root_dir=settings.cache_folder,
model_root_dir=self.cache_dir,
)
)
return session

6
machine-learning/test_main.py
View File

@ -928,7 +928,7 @@ class TestOcr:
session=ort_session.return_value,
rec_batch_num=6,
rec_img_shape=(3, 48, 320),
model_root_dir=settings.cache_folder,
model_root_dir=text_recognizer.cache_dir,
)
)
@ -946,7 +946,7 @@ class TestOcr:
session=ort_session.return_value,
rec_batch_num=4,
rec_img_shape=(3, 48, 320),
model_root_dir=settings.cache_folder,
model_root_dir=text_recognizer.cache_dir,
)
)
@ -966,7 +966,7 @@ class TestOcr:
session=ort_session.return_value,
rec_batch_num=6,
rec_img_shape=(3, 48, 320),
model_root_dir=settings.cache_folder,
model_root_dir=text_recognizer.cache_dir,
)
)