diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py
index baa89be130..fa421a4c23 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 674f52ee35..a964cd28c9 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()