diff --git a/tools/infer/predict_system.py b/tools/infer/predict_system.py index 0265e2a70a4e51d276abf3b6cfa8d1438afc6e48..b4e316d6a5edf464abd846ad2129f5373fc2a36f 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)