From d489d3c27dbd592a7ad761da6b7e7ee868061499 Mon Sep 17 00:00:00 2001
From: MissPenguin <lichenxia1991@163.com>
Date: Fri, 18 Dec 2020 03:40:22 +0000
Subject: [PATCH] fix sast inference bug

---
 tools/infer/predict_det.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py
index ba0adaee..820d3413 100755
--- a/tools/infer/predict_det.py
+++ b/tools/infer/predict_det.py
@@ -70,6 +70,9 @@ class TextDetector(object):
             postprocess_params["cover_thresh"] = args.det_east_cover_thresh
             postprocess_params["nms_thresh"] = args.det_east_nms_thresh
         elif self.det_algorithm == "SAST":
+            pre_process_list[0] = {
+                'DetResizeForTest': {'resize_long': args.det_limit_side_len}
+            }
             postprocess_params['name'] = 'SASTPostProcess'
             postprocess_params["score_thresh"] = args.det_sast_score_thresh
             postprocess_params["nms_thresh"] = args.det_sast_nms_thresh
-- 
GitLab