NeoAnd/Docling
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(ocr): expose rec_keys_path in RapidOcrOptions to support custom dictionaries (#786)

Browse Source 
* Expose `rec_keys_path` in RapidOcrOptions to support custom dictionaries

- Added `rec_keys_path` to `RapidOcrOptions` to align with RapidOCR's capability to use custom character dictionaries.
- Passed `rec_keys_path` to `RapidOcrModel` initialization, ensuring the recognition model can load the correct dictionary (e.g., for Latin characters).

Signed-off-by: Yorick Terweijden <yorick@spread.ai>

* style(rapidocr-options): fix alignment of `rec_keys_path` comment

Adjusted the alignment of the comment for `rec_keys_path` to maintain consistent formatting. No functional changes were made.

Signed-off-by: Yorick Terweijden <yorick@spread.ai>

---------

Signed-off-by: Yorick Terweijden <yorick@spread.ai>
This commit is contained in:
Yorick Terweijden 2025-01-27 14:38:15 +02:00 committed by GitHub
parent 9022c6d855
commit 53327552e8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
docling/datamodel/pipeline_options.py
View File

@ -119,6 +119,7 @@ class RapidOcrOptions(OcrOptions):
det_model_path: Optional[str] = None # same default as rapidocr det_model_path: Optional[str] = None # same default as rapidocr
cls_model_path: Optional[str] = None # same default as rapidocr cls_model_path: Optional[str] = None # same default as rapidocr
rec_model_path: Optional[str] = None # same default as rapidocr rec_model_path: Optional[str] = None # same default as rapidocr
rec_keys_path: Optional[str] = None # same default as rapidocr
model_config = ConfigDict( model_config = ConfigDict(
extra="forbid", extra="forbid",

1
docling/models/rapid_ocr_model.py
View File

@ -59,6 +59,7 @@ class RapidOcrModel(BaseOcrModel):
det_model_path=self.options.det_model_path, det_model_path=self.options.det_model_path,
cls_model_path=self.options.cls_model_path, cls_model_path=self.options.cls_model_path,
rec_model_path=self.options.rec_model_path, rec_model_path=self.options.rec_model_path,
rec_keys_path=self.options.rec_keys_path,
) )
def __call__( def __call__(