mirrors
/
PaddleOCR
mirror of https://github.com/PaddlePaddle/PaddleOCR.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

rever 'limit maximum resizing ratio' (#9449)

pull/9439/head^2
xiaoting 2023-03-15 17:34:37 +08:00 committed by GitHub
parent 9c22490f95
commit 2be9fe763e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
ppocr/data/imaug/rec_img_aug.py
View File

@ -109,9 +109,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,
@ -185,9 +184,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,
@ -572,7 +570,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