"vscode:/vscode.git/clone" did not exist on "a1cb106459da4c595d22c04e026d7169d8dcfd2b"
Commit a2a08583 authored by dyning's avatar dyning
Browse files

use enable_memory_optim for cpu and gpu, add paddlehelp QRcode

parent 6487937a
...@@ -25,7 +25,6 @@ import sys ...@@ -25,7 +25,6 @@ import sys
__dir__ = os.path.dirname(__file__) __dir__ = os.path.dirname(__file__)
sys.path.append(__dir__) sys.path.append(__dir__)
sys.path.append(os.path.join(__dir__, '..')) sys.path.append(os.path.join(__dir__, '..'))
import lanms
class EASTPostPocess(object): class EASTPostPocess(object):
...@@ -79,6 +78,7 @@ class EASTPostPocess(object): ...@@ -79,6 +78,7 @@ class EASTPostPocess(object):
boxes[:, :8] = text_box_restored.reshape((-1, 8)) boxes[:, :8] = text_box_restored.reshape((-1, 8))
boxes[:, 8] = score_map[xy_text[:, 0], xy_text[:, 1]] boxes[:, 8] = score_map[xy_text[:, 0], xy_text[:, 1]]
if self.is_python35: if self.is_python35:
import lanms
boxes = lanms.merge_quadrangle_n9(boxes, nms_thresh) boxes = lanms.merge_quadrangle_n9(boxes, nms_thresh)
else: else:
boxes = nms_locality(boxes.astype(np.float64), nms_thresh) boxes = nms_locality(boxes.astype(np.float64), nms_thresh)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment