Commit 417f76b0 authored by littletomatodonkey's avatar littletomatodonkey
Browse files

fix duplicate thres

parent b5f86130
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment