From 70f87eee237931455f4669c41418f0a89d76a54c Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Thu, 19 Nov 2020 14:32:34 +0800 Subject: [PATCH] fix typo and print --- setup.py | 2 +- tools/infer/predict_det.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/setup.py b/setup.py index 2cea853d..eb6f2d65 100644 --- a/setup.py +++ b/setup.py @@ -15,7 +15,7 @@ from setuptools import setup from io import open -with open('requirments.txt', encoding="utf-8-sig") as f: +with open('requirements.txt', encoding="utf-8-sig") as f: requirements = f.readlines() requirements.append('tqdm') diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py index 60d67c96..b022bc07 100755 --- a/tools/infer/predict_det.py +++ b/tools/infer/predict_det.py @@ -191,10 +191,12 @@ if __name__ == "__main__": if count > 0: total_time += elapse count += 1 - print("Predict time of %s:" % image_file, elapse) + logger.info("Predict time of %s:" % image_file, elapse) src_im = utility.draw_text_det_res(dt_boxes, image_file) 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)) -- GitLab