Skip to content

GitLab

Commit 42853ff0 authored by LDOUBLEV's avatar LDOUBLEV
Browse files

save visualized image

parent b2260182
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tools/infer/predict_det.py
View file @ 42853ff0
...@@ -250,7 +250,7 @@ if __name__ == "__main__": ...@@ -250,7 +250,7 @@ if __name__ == "__main__":
img_name_pure = os.path.split(image_file)[-1] img_name_pure = os.path.split(image_file)[-1]
img_path = os.path.join(draw_img_save, img_path = os.path.join(draw_img_save,
"det_res_{}".format(img_name_pure)) "det_res_{}".format(img_name_pure))
cv2.imwrite(img_path, src_im)
logger.info("The visualized image saved in {}".format(img_path)) logger.info("The visualized image saved in {}".format(img_path))
# print the information about memory and time-spent # print the information about memory and time-spent
if args.benchmark: if args.benchmark:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment