From 4843bccca88551d60c4860d5c2643fbc0f5b0e9a Mon Sep 17 00:00:00 2001 From: andyjpaddle Date: Mon, 15 Aug 2022 06:06:58 +0000 Subject: [PATCH] fix sar infer shape --- tools/infer/predict_rec.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/infer/predict_rec.py b/tools/infer/predict_rec.py index 3664ef2c..5b3e1147 100755 --- a/tools/infer/predict_rec.py +++ b/tools/infer/predict_rec.py @@ -266,7 +266,7 @@ class TextRecognizer(object): for beg_img_no in range(0, img_num, batch_num): end_img_no = min(img_num, beg_img_no + batch_num) norm_img_batch = [] - imgC, imgH, imgW = self.rec_image_shape + imgC, imgH, imgW = self.rec_image_shape[:3] max_wh_ratio = imgW / imgH # max_wh_ratio = 0 for ino in range(beg_img_no, end_img_no): -- GitLab