Skip to content

Commit

Permalink
use enable_memory_optim for cpu and gpu, add paddlehelp QRcode
Browse files Browse the repository at this point in the history
  • Loading branch information
dyning committed Jun 23, 2020
1 parent 6487937 commit a2a0858
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ppocr/postprocess/east_postprocess.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,6 @@
__dir__ = os.path.dirname(__file__)
sys.path.append(__dir__)
sys.path.append(os.path.join(__dir__, '..'))
import lanms


class EASTPostPocess(object):
Expand Down Expand Up @@ -79,6 +78,7 @@ def detect(self,
boxes[:, :8] = text_box_restored.reshape((-1, 8))
boxes[:, 8] = score_map[xy_text[:, 0], xy_text[:, 1]]
if self.is_python35:
import lanms
boxes = lanms.merge_quadrangle_n9(boxes, nms_thresh)
else:
boxes = nms_locality(boxes.astype(np.float64), nms_thresh)
Expand Down

0 comments on commit a2a0858

Please sign in to comment.