From 4e427410a021de82936247d6981ae2905541a6dc Mon Sep 17 00:00:00 2001
From: dyning <duyuning@baidu.com>
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
-- 
GitLab