3D U-Net TensorFlow workspace for BraTS 2020

openfl-workspace/tf_3dunet_brats/src/dataloader.py

@@ -36,7 +36,16 @@ def __init__(self, crop_dim,
 
         self.create_file_list()
 
-        self.ds_train, self.ds_val, self.ds_test = self.get_dataset()
+        if self.num_files > 0:
+
+            self.ds_train, self.ds_val, self.ds_test = self.get_dataset()
+
+        else:
+
+            self.ds_train = None
+            self.ds_val = None
+            self.ds_test = None
+            raise ValueError(f'ERROR: No BraTS datafiles found under directory {self.data_path}')
 
     def create_file_list(self):
         """