yujunjun/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-12-27 15:08:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

add config.enable_memory_optim()

Browse Source
This commit is contained in:
MissPenguin 2020-06-23 11:46:45 +08:00 committed by GitHub
parent f051be8657
commit 160bb06e70
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tools/infer/utility.py
View File

@ -88,6 +88,7 @@ def create_predictor(args, mode):
if args.use_gpu:
config.enable_use_gpu(args.gpu_mem, 0)
config.enable_memory_optim()
else:
config.disable_gpu()
@ -310,4 +311,4 @@ if __name__ == '__main__':
new_img = draw_ocr(image, boxes, txts, scores, draw_txt=True)
cv2.imwrite(img_name, new_img)
cv2.imwrite(img_name, new_img)