diff --git a/tools/export_model.py b/tools/export_model.py index 0bd06b98dcacc06893becbefacbea198c432bc39..7630a77edccc582b07e01f7e585399c65bfc046c 100644 --- a/tools/export_model.py +++ b/tools/export_model.py @@ -15,6 +15,7 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function +from ppocr.utils.utility import enable_static_mode import os import sys @@ -43,7 +44,6 @@ logger = initial_logger() from ppocr.utils.save_load import init_model - def main(): startup_prog, eval_program, place, config, _ = program.preprocess() @@ -72,4 +72,5 @@ def main(): if __name__ == '__main__': + enable_static_mode() main() diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py index 12966718d3e135bb3400522898187c4f1addfdc1..21f9bf90b9546dfb86266a07683d4af1200d53fa 100755 --- a/tools/infer/predict_det.py +++ b/tools/infer/predict_det.py @@ -191,10 +191,13 @@ if __name__ == "__main__": if count > 0: total_time += elapse count += 1 - print("Predict time of %s:" % image_file, elapse) + logger.info("The predicted time of img: {} is {}:".format(image_file, + elapse)) src_im = utility.draw_text_det_res(dt_boxes, image_file) - img_name_pure = os.path.split(image_file)[-1] # image_file.split("/")[-1] - img_path = os.path.join(draw_img_save, "det_res_%s" % img_name_pure) - cv2.imwrite(img_path, src_im) + img_name_pure = image_file.split("/")[-1] + cv2.imwrite( + os.path.join(draw_img_save, "det_res_%s" % img_name_pure), src_im) + logger.info("The visualized img saved in {}".format( + os.path.join(draw_img_save, "det_res_%s" % img_name_pure))) if count > 1: - print("Avg Time:", total_time / (count - 1)) + logger.info("Avg Time:", total_time / (count - 1))