From fb38d0be187dc82e122bc6be3d1c37c57af57d21 Mon Sep 17 00:00:00 2001
From: Alexnet <44656802+shiyunalex@users.noreply.github.com>
Date: Wed, 15 Nov 2023 10:02:03 +0800
Subject: [PATCH] fix SAR inference, when batch size>1, norm_img_batch and
 valid_ratios shapes do not match (#11238)

---
 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 12088b2dc..b47e0c64d 100755
--- a/tools/infer/predict_rec.py
+++ b/tools/infer/predict_rec.py
@@ -584,7 +584,7 @@ class TextRecognizer(object):
                 inputs = [
                     norm_img_batch,
                     np.array(
-                        [valid_ratios], dtype=np.float32),
+                        [valid_ratios], dtype=np.float32).T,
                 ]
                 if self.use_onnx:
                     input_dict = {}