Update dice_correlation.py

neurodata · Feb 4, 2023 · 5a5e746 · 5a5e746
1 parent f68a956
commit 5a5e746
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/scripts/dice_correlation.py b/scripts/dice_correlation.py
@@ -7,7 +7,7 @@
 from math import floor
 
 
-def dice_roi(input_dir, output_dir, atlas1, atlas2):
+def dice_roi(input_dir, output_dir, atlas1, atlas2, verbose=True):
     """Calculates the dice coefficient for every ROI combination from atlas1 and atlas2
 
     Parameters
@@ -48,7 +48,8 @@ def dice_roi(input_dir, output_dir, atlas1, atlas2):
 
             Dice[int(i)][int(j)]=float(dice)
 
-            print(f'Dice coefficient for {yname} {i} of {max_y}, {xname} {j} of {max_x} = {dice}')
+            if verbose:
+                print(f'Dice coefficient for {yname} {i} of {max_y}, {xname} {j} of {max_x} = {dice}')
 
             if dice > 1 or dice < 0:
                 raise ValueError(f"Dice coefficient is greater than 1 or less than 0 ({dice}) at atlas1: {val1}, atlas2: {val2}")
@@ -151,4 +152,4 @@ def main():
                 Dice_matrix, ylabels, xlabels = dice_roi(input_dir,output_dir,atlases[i],atlases[j])
 
 if __name__ == "__main__":
-    main()
+    main()