Skip to content

GitLab

"tests/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "64b3e0f5390728f62887be7820a5e2724d0fb419"
Commit 9b4d77dc authored by myhloli's avatar myhloli
Browse files

fix(lite_model): Adapt iite Mode to the Hybrid OCR Mode in Version 0.10

parent 89c7bd04
Hide whitespace changes
Inline Side-by-side
Showing with 23 additions and 3 deletions
magic_pdf/model/pp_structure_v2.py
View file @ 9b4d77dc
...@@ -18,11 +18,31 @@ def region_to_bbox(region): ...@@ -18,11 +18,31 @@ def region_to_bbox(region):
class CustomPaddleModel: class CustomPaddleModel:
def __init__(self, ocr: bool = False, show_log: bool = False, lang=None): def __init__(self,
ocr: bool = False,
show_log: bool = False,
lang=None,
det_db_box_thresh=0.3,
use_dilation=True,
det_db_unclip_ratio=1.8
):
if lang is not None: if lang is not None:
self.model = PPStructure(table=False, ocr=ocr, show_log=show_log, lang=lang) self.model = PPStructure(table=False,
ocr=True,
show_log=show_log,
lang=lang,
det_db_box_thresh=det_db_box_thresh,
use_dilation=use_dilation,
det_db_unclip_ratio=det_db_unclip_ratio,
)
else: else:
self.model = PPStructure(table=False, ocr=ocr, show_log=show_log) self.model = PPStructure(table=False,
ocr=True,
show_log=show_log,
det_db_box_thresh=det_db_box_thresh,
use_dilation=use_dilation,
det_db_unclip_ratio=det_db_unclip_ratio,
)
def __call__(self, img): def __call__(self, img):
try: try:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment