From 4e427410a021de82936247d6981ae2905541a6dc Mon Sep 17 00:00:00 2001 From: dyning Date: Tue, 14 Jul 2020 14:39:00 +0800 Subject: [PATCH] add parameter enable_mkldnn --- tools/infer/predict_system.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/infer/predict_system.py b/tools/infer/predict_system.py index d2e41054..1b7aff1c 100755 --- a/tools/infer/predict_system.py +++ b/tools/infer/predict_system.py @@ -125,7 +125,7 @@ def main(args): continue starttime = time.time() tackle_img_num += 1 - if not args.use_gpu and tackle_img_num % 30 == 0: + if not args.use_gpu and args.enable_mkldnn and tackle_img_num % 30 == 0: text_sys = TextSystem(args) dt_boxes, rec_res = text_sys(img) elapse = time.time() - starttime