forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #940 from littletomatodonkey/sta/fix_tia 

fix tia exception
This commit is contained in:
littletomatodonkey 2020-10-15 20:06:40 +08:00 committed by GitHub
parent 2177fcdcdb 1b7a16a8b9
commit f4986cb14e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
ppocr/data/rec/img_tools.py
View File

@ -309,16 +309,28 @@ def warp(img, ang):
if config.distort: if config.distort:
img_height, img_width = img.shape[0:2] img_height, img_width = img.shape[0:2]
if random.random() <= prob and img_height >= 20 and img_width >= 20: if random.random() <= prob and img_height >= 20 and img_width >= 20:
new_img = tia_distort(new_img, random.randint(3, 6)) try:
new_img = tia_distort(new_img, random.randint(3, 6))
except:
logger.warning(
"Exception occured during tia_distort, pass it...")
if config.stretch: if config.stretch:
img_height, img_width = img.shape[0:2] img_height, img_width = img.shape[0:2]
if random.random() <= prob and img_height >= 20 and img_width >= 20: if random.random() <= prob and img_height >= 20 and img_width >= 20:
new_img = tia_stretch(new_img, random.randint(3, 6)) try:
new_img = tia_stretch(new_img, random.randint(3, 6))
except:
logger.warning(
"Exception occured during tia_stretch, pass it...")
if config.perspective: if config.perspective:
if random.random() <= prob: if random.random() <= prob:
new_img = tia_perspective(new_img) try:
new_img = tia_perspective(new_img)
except:
logger.warning(
"Exception occured during tia_perspective, pass it...")
if config.crop: if config.crop:
img_height, img_width = img.shape[0:2] img_height, img_width = img.shape[0:2]