add parameter enable_mkldnn

5 years ago · 4e427410a0
parent 814dc2fa37
commit 4e427410a0
1 changed files with 1 additions and 1 deletions
--- 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