rever 'limit maximum resizing ratio' (#9448)

This commit is contained in:
xiaoting
2023-03-15 18:49:45 +08:00
committed by GitHub
parent 77b05c2add
commit 1e8b7eb510
+5 -7
View File
@@ -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