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

feat: Add Easyocr parameter recog_network (#613)

Browse Source 
* Update easyocr_model.py

Added this line of code to get recog_network of easyocr parameter
recog_network = self.options.recog_network

Signed-off-by: itsainii <aininawawii@gmail.com>

* Update pipeline_options.py

Added this line in EasyOcrOptions function

recog_network: Optional[str] = 'standard'

Signed-off-by: itsainii <aininawawii@gmail.com>

* Add Easyocr recog_network parameter

Signed-off-by: itsainii <aininawawii@gmail.com>

---------

Signed-off-by: itsainii <aininawawii@gmail.com>
This commit is contained in:
itsainii 2024-12-17 16:47:18 +08:00 committed by GitHub
parent 3bb3bf5715
commit 3b53bd38c8
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

@ -140,6 +140,7 @@ class EasyOcrOptions(OcrOptions):
use_gpu: Optional[bool] = None use_gpu: Optional[bool] = None
model_storage_directory: Optional[str] = None model_storage_directory: Optional[str] = None
recog_network: Optional[str] = "standard"
download_enabled: bool = True download_enabled: bool = True
model_config = ConfigDict( model_config = ConfigDict(

1
docling/models/easyocr_model.py
View File

@ -66,6 +66,7 @@ class EasyOcrModel(BaseOcrModel):
lang_list=self.options.lang, lang_list=self.options.lang,
gpu=use_gpu, gpu=use_gpu,
model_storage_directory=self.options.model_storage_directory, model_storage_directory=self.options.model_storage_directory,
recog_network=self.options.recog_network,
download_enabled=self.options.download_enabled, download_enabled=self.options.download_enabled,
verbose=False, verbose=False,
) )