limit maximum resizing ratio

2022-08-26 22:17:23 +08:00 · 2022-08-26 22:17:23 +08:00 · 4af5f63d44
parent 18e80f9b02
commit 4af5f63d44
1 changed files with 1 additions and 1 deletions
--- a/ppocr/data/imaug/rec_img_aug.py
+++ b/ppocr/data/imaug/rec_img_aug.py
@ -502,7 +502,7 @@ def resize_norm_img_chinese(img, image_shape):
    max_wh_ratio = imgW * 1.0 / imgH
    h, w = img.shape[0], img.shape[1]
    ratio = w * 1.0 / h
-    max_wh_ratio = max(max_wh_ratio, ratio)
+    max_wh_ratio = min(max(max_wh_ratio, ratio), max_wh_ratio)
    imgW = int(imgH * max_wh_ratio)
    if math.ceil(imgH * ratio) > imgW:
        resized_w = imgW