fix SAR inference, when batch size>1, norm_img_batch and valid_ratios shapes do not match (#11238)

pull/11254/head
Alexnet 2023-11-15 10:02:03 +08:00 committed by GitHub
parent 73c08b7e4c
commit fb38d0be18
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -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 = {}