Fix path to Salinas dataset (closes #17)

nshaud · Dec 3, 2020 · 176e582 · 176e582
1 parent c672648
commit 176e582
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/datasets.py b/datasets.py
@@ -182,11 +182,11 @@ def get_dataset(dataset_name, target_folder="./", datasets=DATASETS_CONFIG):
         ignored_labels = [0]
 
     elif dataset_name == "Salinas":
-        img = open_file(folder + "Salinas.mat")["Salinas_corrected"]
+        img = open_file(folder + "Salinas_corrected.mat")["salinas_corrected"]
 
         rgb_bands = (43, 21, 11)  # AVIRIS sensor
 
-        gt = open_file(folder + "Salinas_gt.mat")["Salinas_gt"]
+        gt = open_file(folder + "Salinas_gt.mat")["salinas_gt"]
 
         label_values = [
             "Undefined",