From 8223759fdff18863cdf413ec0d655d1b20d80c29 Mon Sep 17 00:00:00 2001 From: liuyebo <1515783401@qq.com> Date: Tue, 5 Nov 2024 17:18:12 +0800 Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4=E8=87=AA=E5=8A=A8=E8=AF=86?= =?UTF-8?q?=E5=88=AB=E7=9A=84=E6=96=87=E5=AD=97=E8=AF=86=E5=88=AB=E6=A8=A1?= =?UTF-8?q?=E5=9E=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- photo_review/__init__.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/photo_review/__init__.py b/photo_review/__init__.py index 23a4d76..5b8be9f 100644 --- a/photo_review/__init__.py +++ b/photo_review/__init__.py @@ -101,4 +101,14 @@ DISCHARGE_IE = Taskflow('information_extraction', schema=DISCHARGE_RECORD_SCHEMA COST_IE = Taskflow('information_extraction', schema=COST_LIST_SCHEMA, model='uie-x-base', device_id=1, task_path='model/cost_list_model', layout_analysis=LAYOUT_ANALYSIS, precision='fp16') -OCR = PaddleOCR(use_angle_cls=False, show_log=False, gpu_id=1, det_db_box_thresh=0.3) +OCR = PaddleOCR( + gpu_id=1, + use_angle_cls=False, + show_log=False, + det_db_thresh=0.1, + det_db_box_thresh=0.3, + det_limit_side_len=1248, + drop_score=0.3, + rec_model_dir='model/ocr/openatom_rec_repsvtr_ch_infer', + rec_algorithm='SVTR_LCNet', +)