Better handling of wrong (non existing) path

gmberton · Jan 3, 2023 · dfc524c · dfc524c
1 parent eab6e11
commit dfc524c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/datasets/train_dataset.py b/datasets/train_dataset.py
@@ -104,6 +104,9 @@ def __len__(self):
     def initialize(dataset_folder, M, N, alpha, L, min_images_per_class, filename):
         logging.debug(f"Searching training images in {dataset_folder}")
 
+        if not os.path.exists(dataset_folder):
+            raise FileNotFoundError(f"Folder {dataset_folder} does not exist")
+
         images_paths = sorted(glob(f"{dataset_folder}/**/*.jpg", recursive=True))
         logging.debug(f"Found {len(images_paths)} images")