Merge pull request #39 from sarthakpati/master

Author: Geeks-Sid

GANDLF/training_loop.py

@@ -167,7 +167,9 @@ def trainingLoop(trainingDataFromPickle, validationDataFromPickle, headers, devi
                     dice_weights_dict[i] = dice_weights_dict[i] + currentNumber # class-specific non-zero voxels
                     total_nonZeroVoxels = total_nonZeroVoxels + currentNumber # total number of non-zero voxels to be considered
 
-            
+            if total_nonZeroVoxels == 0:
+                raise RuntimeError('Trying to train on data where every label mask is background class only.')
+
             # dice_weights_dict_temp = deepcopy(dice_weights_dict)
             dice_weights_dict = {k: (v / total_nonZeroVoxels) for k, v in dice_weights_dict.items()} # divide each dice value by total nonzero
             dice_penalty_dict = deepcopy(dice_weights_dict) # deep copy so that both values are preserved

setup.py

@@ -34,7 +34,7 @@
 
 setup(
   name='GANDLF',
-  version='0.0.8-dev', # NR: non-release; this should be changed when tagging
+  version='0.0.8', # NR: non-release; this should be changed when tagging
   author="Ujjwal Baid, Megh Bhalerao, Caleb Grenko, Sarthak Pati, Siddhesh Thakur", # alphabetical order
   author_email='[email protected]',
   python_requires='>=3.6',