From 8ceae0bb9efccce380f3f0899637c8df6696f7a7 Mon Sep 17 00:00:00 2001 From: hungcao0402 <79310692+hungcao0402@users.noreply.github.com> Date: Mon, 29 Nov 2021 01:54:31 +0700 Subject: [PATCH] Update predict_det.py I found that when I export model SAST to predict image, that made difference predict bounding box between inference model and train model --- tools/infer/predict_det.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py index 6347ca6d..ff3fb0d5 100755 --- a/tools/infer/predict_det.py +++ b/tools/infer/predict_det.py @@ -74,7 +74,7 @@ class TextDetector(object): elif self.det_algorithm == "SAST": pre_process_list[0] = { 'DetResizeForTest': { - 'resize_long': args.det_limit_side_len + 'resize_long': 1536 } } postprocess_params['name'] = 'SASTPostProcess' -- GitLab