.

criteria.py

@@ -66,6 +66,7 @@ def __init__(self, *args, **kwargs):
         super(Hausdorff_Distance, self).__init__()
 
     def forward(self, y_pred, y_true):
+        # Not Accurate Hausdorff
         y_pred = y_pred.squeeze().detach().cpu().numpy()
         y_true = y_true.squeeze().detach().cpu().numpy()
         hausdorff_distance = []

dataset.py

@@ -43,10 +43,7 @@ def __del__(self) -> None:
 
 # Read from raw data is easier than read from hdf5 file
 class BraTSDataset_from_nii(Dataset):
-
-
     # https://www.synapse.org/#!Synapse:syn52939291/wiki/625694
-
     def __init__(self, subjects_list : list) -> None:
         super().__init__()
         self.nii_path_list = [[os.path.join(each_subject, file_name) for file_name in os.listdir(each_subject) ] for each_subject in subjects_list]