Merge pull request #3809 from tink2123/fix_infer_shape

fix infer shape
2021-08-26 13:45:36 +08:00 · 2021-08-26 13:45:36 +08:00 · 28a40efe27
parent d77db841b0 22b57d632d
commit 28a40efe27
1 changed files with 2 additions and 2 deletions
--- a/tools/infer/predict_rec.py
+++ b/tools/infer/predict_rec.py
@ -88,8 +88,8 @@ class TextRecognizer(object):
    def resize_norm_img(self, img, max_wh_ratio):
        imgC, imgH, imgW = self.rec_image_shape
        assert imgC == img.shape[2]
-        if self.character_type == "ch":
-            imgW = int((32 * max_wh_ratio))
+        max_wh_ratio = max(max_wh_ratio, imgW / imgH)
+        imgW = int((32 * max_wh_ratio))
        h, w = img.shape[:2]
        ratio = w / float(h)
        if math.ceil(imgH * ratio) > imgW: