diff --git a/ppocr/data/imaug/rec_img_aug.py b/ppocr/data/imaug/rec_img_aug.py index c8883cb380ca7e1fbaa9a27e099ceaf75279c295..9ced5e4a1fc3eff258875283df1b7361f1eae0b7 100644 --- a/ppocr/data/imaug/rec_img_aug.py +++ b/ppocr/data/imaug/rec_img_aug.py @@ -112,9 +112,8 @@ class ABINetRecAug(object): scale=(0.5, 2.), shear=(45, 15), distortion=0.5, - p=geometry_p), - CVDeterioration( - var=20, degrees=6, factor=4, p=deterioration_p), + p=geometry_p), CVDeterioration( + var=20, degrees=6, factor=4, p=deterioration_p), CVColorJitter( brightness=0.5, contrast=0.5, @@ -188,9 +187,8 @@ class SVTRRecAug(object): scale=(0.5, 2.), shear=(45, 15), distortion=0.5, - p=geometry_p), - SVTRDeterioration( - var=20, degrees=6, factor=4, p=deterioration_p), + p=geometry_p), SVTRDeterioration( + var=20, degrees=6, factor=4, p=deterioration_p), CVColorJitter( brightness=0.5, contrast=0.5, @@ -575,7 +573,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 = min(max(max_wh_ratio, ratio), max_wh_ratio) + max_wh_ratio = max(max_wh_ratio, ratio) imgW = int(imgH * max_wh_ratio) if math.ceil(imgH * ratio) > imgW: resized_w = imgW