diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py index f5ea0504f97f3e40853d431061f7086653f2628e..82492abcfa9f17e0dbb94a8331237307e74ea9dc 100755 --- a/tools/infer/predict_det.py +++ b/tools/infer/predict_det.py @@ -39,7 +39,10 @@ class TextDetector(object): self.args = args self.det_algorithm = args.det_algorithm pre_process_list = [{ - 'DetResizeForTest': None + 'DetResizeForTest': { + 'limit_side_len': args.det_limit_side_len, + 'limit_type': args.det_limit_type + } }, { 'NormalizeImage': { 'std': [0.229, 0.224, 0.225], @@ -159,7 +162,7 @@ class TextDetector(object): shape_list = np.expand_dims(shape_list, axis=0) img = img.copy() starttime = time.time() - + print(img.shape) self.input_tensor.copy_from_cpu(img) self.predictor.run() outputs = []