fix sar infer shape

2022-08-15 06:06:58 +00:00 · 2022-08-15 06:06:58 +00:00 · 4843bccca8
parent c89bc39789
commit 4843bccca8
1 changed files with 1 additions and 1 deletions
--- 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):