Skip to content

GitLab

Commit 4843bccc authored by andyjpaddle's avatar andyjpaddle
Browse files

fix sar infer shape

parent c89bc397
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tools/infer/predict_rec.py
View file @ 4843bccc
...@@ -266,7 +266,7 @@ class TextRecognizer(object): ...@@ -266,7 +266,7 @@ class TextRecognizer(object):
for beg_img_no in range(0, img_num, batch_num): for beg_img_no in range(0, img_num, batch_num):
end_img_no = min(img_num, beg_img_no + batch_num) end_img_no = min(img_num, beg_img_no + batch_num)
norm_img_batch = [] 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 = imgW / imgH
# max_wh_ratio = 0 # max_wh_ratio = 0
for ino in range(beg_img_no, end_img_no): for ino in range(beg_img_no, end_img_no):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment