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

Merge pull request #8 from LDOUBLEV/fixocr 

update det process code, +32 -> -32
This commit is contained in:
dyning 2020-05-13 22:54:07 +08:00 committed by GitHub
parent 5022503081 2a298fa4cf
commit c01e32672f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ppocr/data/det/db_process.py
View File

@ -143,12 +143,16 @@ class DBProcessTest(object):
resize_w = int(resize_w * ratio) resize_w = int(resize_w * ratio)
if resize_h % 32 == 0: if resize_h % 32 == 0:
resize_h = resize_h resize_h = resize_h
elif resize_h // 32 <= 1:
resize_h = 32
else: else:
resize_h = (resize_h // 32 + 1) * 32 resize_h = (resize_h // 32 - 1) * 32
if resize_w % 32 == 0: if resize_w % 32 == 0:
resize_w = resize_w resize_w = resize_w
elif resize_w // 32 <= 1:
resize_w = 32
else: else:
resize_w = (resize_w // 32 + 1) * 32 resize_w = (resize_w // 32 - 1) * 32
try: try:
if int(resize_w) <= 0 or int(resize_h) <= 0: if int(resize_w) <= 0 or int(resize_h) <= 0:
return None, (None, None) return None, (None, None)

4
ppocr/data/det/east_process.py
View File

@ -485,10 +485,14 @@ class EASTProcessTest(object):
resize_w = int(resize_w * ratio) resize_w = int(resize_w * ratio)
if resize_h % 32 == 0: if resize_h % 32 == 0:
resize_h = resize_h resize_h = resize_h
elif resize_h // 32 <= 1:
resize_h = 32
else: else:
resize_h = (resize_h // 32 - 1) * 32 resize_h = (resize_h // 32 - 1) * 32
if resize_w % 32 == 0: if resize_w % 32 == 0:
resize_w = resize_w resize_w = resize_w
elif resize_w // 32 <= 1:
resize_w = 32
else: else:
resize_w = (resize_w // 32 - 1) * 32 resize_w = (resize_w // 32 - 1) * 32
im = cv2.resize(im, (int(resize_w), int(resize_h))) im = cv2.resize(im, (int(resize_w), int(resize_h)))