diff --git a/tools/infer/predict_system.py b/tools/infer/predict_system.py index e96a1934bd86dfea45257a6ffab42f40cecf85ad..65478b6964c36a88507cd6c6895fdb74984f3eaa 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))