yujunjun/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-11-04 03:39:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

opt east post

Browse Source
This commit is contained in:
WenmuZhou 2021-11-29 02:48:09 +00:00
parent 2c003d5645
commit 389e315ae0
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
ppocr/postprocess/east_postprocess.py
View File

@ -60,11 +60,6 @@ class EASTPostProcess(object):
"""
restore text boxes from score map and geo map
"""
try:
import lanms
except:
raise Exception(
'you should install lanms by pip3 install lanms-nova')
score_map = score_map[0]
geo_map = np.swapaxes(geo_map, 1, 0)
@ -81,8 +76,15 @@ class EASTPostProcess(object):
boxes = np.zeros((text_box_restored.shape[0], 9), dtype=np.float32)
boxes[:, :8] = text_box_restored.reshape((-1, 8))
boxes[:, 8] = score_map[xy_text[:, 0], xy_text[:, 1]]
try:
import lanms
boxes = lanms.merge_quadrangle_n9(boxes, nms_thresh)
# boxes = nms_locality(boxes.astype(np.float64), nms_thresh)
except:
print(
'you should install lanms by pip3 install lanms-nova to speed up nms_locality'
)
boxes = nms_locality(boxes.astype(np.float64), nms_thresh)
if boxes.shape[0] == 0:
return []
# Here we filter some low score boxes by the average score map,