save visualized image

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:

tools/infer_det.py

@@ -112,4 +112,4 @@ def main():
 
 if __name__ == '__main__':
     config, device, logger, vdl_writer = program.preprocess()
-    main()
+    main()
\ No newline at end of file