Updated detect.py to save images in images/ directory when save_txt is True

detect.py

@@ -108,6 +108,9 @@ def run(
 
     # Directories
     save_dir = increment_path(Path(project) / name, exist_ok=exist_ok)  # increment run
+    (save_dir / "images").mkdir(parents=True, exist_ok=True) if save_txt else None
+    img_save_path = save_dir / "images" if save_txt else save_dir
+    # Images and lables will be saved in different directories if save_txt is true
     (save_dir / "labels" if save_txt else save_dir).mkdir(parents=True, exist_ok=True)  # make dir
 
     # Load model
@@ -184,7 +187,7 @@ def write_to_csv(image_name, prediction, confidence):
                 p, im0, frame = path, im0s.copy(), getattr(dataset, "frame", 0)
 
             p = Path(p)  # to Path
-            save_path = str(save_dir / p.name)  # im.jpg
+            save_path = str(img_save_path / p.name)  # im.jpg
             txt_path = str(save_dir / "labels" / p.stem) + ("" if dataset.mode == "image" else f"_{frame}")  # im.txt
             s += "%gx%g " % im.shape[2:]  # print string
             gn = torch.tensor(im0.shape)[[1, 0, 1, 0]]  # normalization gain whwh