Skip to content

GitLab

Commit b5f86130 authored by littletomatodonkey's avatar littletomatodonkey
Browse files

fix typo

parent 01fd2d0f
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 4 deletions
tools/infer/predict_system.py
View file @ b5f86130
...@@ -93,11 +93,11 @@ class TextSystem(object): ...@@ -93,11 +93,11 @@ class TextSystem(object):
self.draw_crop_rec_res(self.args.crop_res_save_dir, img_crop_list, self.draw_crop_rec_res(self.args.crop_res_save_dir, img_crop_list,
rec_res) rec_res)
filter_boxes, filter_rec_res = [], [] filter_boxes, filter_rec_res = [], []
for box, rec_reuslt in zip(dt_boxes, rec_res): for box, rec_result in zip(dt_boxes, rec_res):
text, score = rec_reuslt text, score = rec_result
if score >= self.drop_score: if score >= self.drop_score:
filter_boxes.append(box) filter_boxes.append(box)
filter_rec_res.append(rec_reuslt) filter_rec_res.append(rec_result)
return filter_boxes, filter_rec_res return filter_boxes, filter_rec_res
...@@ -197,7 +197,6 @@ def main(args): ...@@ -197,7 +197,6 @@ def main(args):
with open(os.path.join(draw_img_save_dir, "system_results.txt"), 'w') as f: with open(os.path.join(draw_img_save_dir, "system_results.txt"), 'w') as f:
f.writelines(save_results) f.writelines(save_results)
f.close()
if __name__ == "__main__": if __name__ == "__main__":
......
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