yujunjun/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-11-02 10:49:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix duplicate thres

Browse Source
This commit is contained in:
littletomatodonkey 2022-02-07 08:23:40 +00:00
parent b5f8613007
commit 417f76b060
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/infer/predict_system.py
View File

@ -164,7 +164,7 @@ def main(args):
res = [{
"transcription": rec_res[idx][0],
"points": np.array(dt_boxes[idx]).astype(np.int32).tolist(),
} for idx in range(len(dt_boxes)) if rec_res[idx][1] >= drop_score]
} for idx in range(len(dt_boxes))]
save_pred = os.path.basename(image_file) + "\t" + json.dumps(
res, ensure_ascii=False) + "\n"
save_results.append(save_pred)