From 59d50c919ca3154696ec740edd86766a2acf86c0 Mon Sep 17 00:00:00 2001 From: ioracion Date: Tue, 25 Jan 2022 14:34:00 +0800 Subject: [PATCH 1/2] Update predict_system.py --- tools/infer/predict_system.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tools/infer/predict_system.py b/tools/infer/predict_system.py index 8d674809a5..16789b81cd 100755 --- a/tools/infer/predict_system.py +++ b/tools/infer/predict_system.py @@ -92,11 +92,11 @@ class TextSystem(object): self.draw_crop_rec_res(self.args.crop_res_save_dir, img_crop_list, rec_res) filter_boxes, filter_rec_res = [], [] - for box, rec_reuslt in zip(dt_boxes, rec_res): - text, score = rec_reuslt + for box, rec_result in zip(dt_boxes, rec_res): + text, score = rec_result if score >= self.drop_score: filter_boxes.append(box) - filter_rec_res.append(rec_reuslt) + filter_rec_res.append(rec_result) return filter_boxes, filter_rec_res From a572737fb6af77710fc869803df06fc963603aea Mon Sep 17 00:00:00 2001 From: ioracion Date: Tue, 25 Jan 2022 14:35:43 +0800 Subject: [PATCH 2/2] Update infer_cls.py --- tools/infer_cls.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/infer_cls.py b/tools/infer_cls.py index 7522e43907..ab6a49120b 100755 --- a/tools/infer_cls.py +++ b/tools/infer_cls.py @@ -73,8 +73,8 @@ def main(): images = paddle.to_tensor(images) preds = model(images) post_result = post_process_class(preds) - for rec_reuslt in post_result: - logger.info('\t result: {}'.format(rec_reuslt)) + for rec_result in post_result: + logger.info('\t result: {}'.format(rec_result)) logger.info("success!")