fix SAR inference, when batch size>1, norm_img_batch and valid_ratios shapes do not match (#11238)
parent
73c08b7e4c
commit
fb38d0be18
|
@ -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 = {}
|
||||
|
|
Loading…
Reference in New Issue