From b8a98bc32e1ed4c770c30c18d1cd19b7716867b7 Mon Sep 17 00:00:00 2001 From: andyjpaddle <87074272+andyjpaddle@users.noreply.github.com> Date: Mon, 15 Aug 2022 14:16:34 +0800 Subject: [PATCH] [CP]Fix sar infer shape (#7199) * update doc for quickstart * update some note * 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