From 87f0aeed705a7b07038d4ec1f4c63b31b0298d24 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Wed, 24 Feb 2021 15:09:37 +0800 Subject: [PATCH 1/2] fix gap between train_model predict and inference --- tools/infer/predict_det.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py index 76c6a447..b14825bd 100755 --- a/tools/infer/predict_det.py +++ b/tools/infer/predict_det.py @@ -39,10 +39,7 @@ class TextDetector(object): self.args = args self.det_algorithm = args.det_algorithm pre_process_list = [{ - 'DetResizeForTest': { - 'limit_side_len': args.det_limit_side_len, - 'limit_type': args.det_limit_type - } + 'DetResizeForTest': None }, { 'NormalizeImage': { 'std': [0.229, 0.224, 0.225], From 85fd65c8f021d5b3e31a1d09a6e2a7bae0dd1dbe Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Wed, 24 Feb 2021 20:20:17 +0800 Subject: [PATCH 2/2] fix typo --- tools/infer_det.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/infer_det.py b/tools/infer_det.py index d890970e..913d617d 100755 --- a/tools/infer_det.py +++ b/tools/infer_det.py @@ -97,7 +97,7 @@ def main(): preds = model(images) post_result = post_process_class(preds, shape_list) boxes = post_result[0]['points'] - # write resule + # write result dt_boxes_json = [] for box in boxes: tmp_json = {"transcription": ""}