Echo/fcb_photo_review
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

同步调整丢弃的阈值

Browse Source
This commit is contained in:
liuyebo
2024-07-25 13:40:00 +08:00
parent bde460dc77
commit a0a9ea9edb
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
visual_model_test/visual_model_test.py
View File

@@ -63,8 +63,9 @@ def visual_model_test(model_type, test_img, task_path, schema):
img["y_offset"] -= offset_y img["y_offset"] -= offset_y
temp_files_paths.append(temp_file.name) 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), parsed_doc = util.get_ocr_layout(
temp_file.name) 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"] # parsed_doc = doc_parser.parse({"doc": temp_file.name})["layout"]
if img["x_offset"] or img["y_offset"]: if img["x_offset"] or img["y_offset"]:
for p in parsed_doc: for p in parsed_doc: