diff --git a/tools/infer/predict_rec.py b/tools/infer/predict_rec.py
index 4e4150c515fc2d0ee4eb7e635cb8c81a467e748f..449f69ed6a22cb12af8a6a2ef8f2eedc1aca087c 100755
--- a/tools/infer/predict_rec.py
+++ b/tools/infer/predict_rec.py
@@ -342,7 +342,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):