fix for non-int subject ids

fabiocarrara · Feb 3, 2022 · f9c3b15 · f9c3b15
1 parent 1d68967
commit f9c3b15
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dataloader.py b/dataloader.py
@@ -141,7 +141,6 @@ def load_datum(datum, x_shape=(128, 128, 1), augment=False):
     return x, y, y2
 
 
-
 def get_loader(dataframe, batch_size=8, shuffle=False, **kwargs):
     categories = dataframe.exp.values
 
@@ -174,6 +173,7 @@ def _load_and_prepare_annotations(dataset_dir):
         return data
 
     dataset = pd.concat([_load_and_prepare_annotations(d) for d in dataset_dirs])
+    dataset['sub'] = dataset['sub'].astype(str)
     return dataset