diff --git a/tools/infer/predict_system.py b/tools/infer/predict_system.py index d2e41054b5c8e6f3b61eaa7fed3547c56f2f68e9..1b7aff1ca8213633b7eff73e13d0ea75eeaedc0b 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