From a0a9ea9edb66c9c473f75820cb5f87e2b52650ed Mon Sep 17 00:00:00 2001 From: liuyebo <1515783401@qq.com> Date: Thu, 25 Jul 2024 13:40:00 +0800 Subject: [PATCH] =?UTF-8?q?=E5=90=8C=E6=AD=A5=E8=B0=83=E6=95=B4=E4=B8=A2?= =?UTF-8?q?=E5=BC=83=E7=9A=84=E9=98=88=E5=80=BC?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- visual_model_test/visual_model_test.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/visual_model_test/visual_model_test.py b/visual_model_test/visual_model_test.py index 9714b5c..d5196b4 100644 --- a/visual_model_test/visual_model_test.py +++ b/visual_model_test/visual_model_test.py @@ -63,8 +63,9 @@ 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=1248), - temp_file.name) + parsed_doc = util.get_ocr_layout( + PaddleOCR(det_db_box_thresh=0.3, det_limit_side_len=1248, drop_score=0.3), + temp_file.name) # parsed_doc = doc_parser.parse({"doc": temp_file.name})["layout"] if img["x_offset"] or img["y_offset"]: for p in parsed_doc: