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', +)