add config.enable_memory_optim()

2026-09-24 23:33:08 +08:00 · 2020-06-23 11:46:45 +08:00
parent f051be8657
commit 160bb06e70
1 changed files with 2 additions and 1 deletions
@@ -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)