diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py
index c5e25903c17378bb992ebdd4fc7ead012528dd46..1b52e717ca1edbbd9ede31260e47ec8973270d3f 100755
--- 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))
diff --git a/tools/infer_det.py b/tools/infer_det.py
index 674f52ee35aab25356ccdbf371f8bac5b52b871a..a964cd28c934504ce79ea4873d3345295c1266e5 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()