From 010533bd570a57daa4eb038c9b155e2a58173707 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Thu, 14 May 2020 14:25:00 +0800 Subject: [PATCH] change the saved path of visualized image --- tools/infer/predict_system.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tools/infer/predict_system.py b/tools/infer/predict_system.py index 83acdbe5..48d3e54e 100755 --- a/tools/infer/predict_system.py +++ b/tools/infer/predict_system.py @@ -132,9 +132,11 @@ if __name__ == "__main__": draw_img = draw_ocr( image, boxes, txts, scores, draw_txt=True, drop_score=0.5) - draw_img_save = "./doc/imgs_results/" + draw_img_save = "./inference_results/" if not os.path.exists(draw_img_save): os.makedirs(draw_img_save) cv2.imwrite( os.path.join(draw_img_save, os.path.basename(image_file)), draw_img) + print("The visualized image saved in {}".format( + os.path.join(draw_img_save, os.path.basename(image_file)))) -- GitLab