From 42853ff0deee518de04917d7bce02ead05bdc52a Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Thu, 24 Jun 2021 12:00:58 +0000 Subject: [PATCH] save visualized image --- tools/infer/predict_det.py | 2 +- tools/infer_det.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py index baa89be1..fa421a4c 100755 --- a/tools/infer/predict_det.py +++ b/tools/infer/predict_det.py @@ -250,7 +250,7 @@ if __name__ == "__main__": 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)) # print the information about memory and time-spent if args.benchmark: diff --git a/tools/infer_det.py b/tools/infer_det.py index 674f52ee..a964cd28 100755 --- a/tools/infer_det.py +++ b/tools/infer_det.py @@ -112,4 +112,4 @@ def main(): if __name__ == '__main__': config, device, logger, vdl_writer = program.preprocess() - main() \ No newline at end of file + main() -- GitLab