diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py index ba0adaee258096ea9970425cc05ca7a8f1cf08c4..820d3413176cd606a32c1c27fa362eca58ee03ab 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