use logger.info replace print
@@ -178,11 +178,12 @@ if __name__ == "__main__":
if count > 0:
total_time += elapse
count += 1
- print("Predict time of {}: {}".format(image_file, elapse))
+ logger.info("Predict time of {}: {}".format(image_file, elapse))
src_im = utility.draw_text_det_res(dt_boxes, image_file)
img_name_pure = os.path.split(image_file)[-1]
img_path = os.path.join(draw_img_save,
"det_res_{}".format(img_name_pure))
cv2.imwrite(img_path, src_im)
+ logger.info("The visualized image saved in {}".format(img_path))
if count > 1:
- print("Avg Time:", total_time / (count - 1))
+ logger.info("Avg Time:", total_time / (count - 1))
