Merge pull request #3185 from LDOUBLEV/fix_vis

save visualized image

Merge pull request #3185 from LDOUBLEV/fix_vis
save visualized image
bc999986 · xiaoting · GitHub · 7363d2f9 · 42853ff0 · bc999986
Unverified Commit bc999986 authored Jun 25, 2021 by xiaoting Committed by GitHub Jun 25, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tools/infer/predict_det.py tools/infer/predict_det.py +1 -1

No files found.
--- a/tools/infer/predict_det.py
+++ b/tools/infer/predict_det.py
@@ -235,5 +235,5 @@ 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))