yujunjun/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-12-26 22:48:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/infer/predict_rec.py
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 = {}