diff --git a/photo_mask/__init__.py b/photo_mask/__init__.py index 1319a73..bc0c009 100644 --- a/photo_mask/__init__.py +++ b/photo_mask/__init__.py @@ -29,4 +29,4 @@ SIMILAR_CHAR = { } # 如果不希望识别出空格,可以设置use_space_char=False。做此项设置一定要测试,2.7.3版本此项设置有bug,会导致识别失败 -OCR = PaddleOCR(use_angle_cls=False, show_log=False, det_db_box_thresh=0.3, det_limit_side_len=2048) +OCR = PaddleOCR(use_angle_cls=False, show_log=False, det_db_box_thresh=0.3, det_limit_side_len=1248) diff --git a/visual_model_test/visual_model_test.py b/visual_model_test/visual_model_test.py index 72e85e7..9714b5c 100644 --- a/visual_model_test/visual_model_test.py +++ b/visual_model_test/visual_model_test.py @@ -63,7 +63,7 @@ def visual_model_test(model_type, test_img, task_path, schema): img["y_offset"] -= offset_y temp_files_paths.append(temp_file.name) - parsed_doc = util.get_ocr_layout(PaddleOCR(det_db_box_thresh=0.3, det_limit_side_len=2048), + parsed_doc = util.get_ocr_layout(PaddleOCR(det_db_box_thresh=0.3, det_limit_side_len=1248), temp_file.name) # parsed_doc = doc_parser.parse({"doc": temp_file.name})["layout"] if img["x_offset"] or img["y_offset"]: