From 814dc2fa37c08d0067e717094b3bca4f0515ed8e Mon Sep 17 00:00:00 2001 From: dyning Date: Tue, 14 Jul 2020 14:35:18 +0800 Subject: [PATCH] add parameter enable_mkldnn --- tools/infer/predict_system.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tools/infer/predict_system.py b/tools/infer/predict_system.py index e96a1934..d2e41054 100755 --- a/tools/infer/predict_system.py +++ b/tools/infer/predict_system.py @@ -117,12 +117,16 @@ def main(args): image_file_list = get_image_file_list(args.image_dir) text_sys = TextSystem(args) is_visualize = True + tackle_img_num = 0 for image_file in image_file_list: img = cv2.imread(image_file) if img is None: logger.info("error in loading image:{}".format(image_file)) continue starttime = time.time() + tackle_img_num += 1 + if not args.use_gpu and tackle_img_num % 30 == 0: + text_sys = TextSystem(args) dt_boxes, rec_res = text_sys(img) elapse = time.time() - starttime print("Predict time of %s: %.3fs" % (image_file, elapse))