From 417f76b06081e15ddc60bc7076a0e4d04051421b Mon Sep 17 00:00:00 2001 From: littletomatodonkey Date: Mon, 7 Feb 2022 08:23:40 +0000 Subject: [PATCH] fix duplicate thres --- tools/infer/predict_system.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/infer/predict_system.py b/tools/infer/predict_system.py index 0265e2a7..b4e316d6 100755 --- a/tools/infer/predict_system.py +++ b/tools/infer/predict_system.py @@ -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) -- GitLab