PytorchConnectomics
diff --git a/‎ves_seg/sample/7-13_clahe.h5‎
26.3 MB b/‎ves_seg/sample/7-13_clahe.h5‎
26.3 MB
diff --git a/‎ves_seg/sample/7-13_mask.h5‎
227 KB b/‎ves_seg/sample/7-13_mask.h5‎
227 KB
diff --git a/‎ves_seg/sample/7-13_ves.h5‎
190 KB b/‎ves_seg/sample/7-13_ves.h5‎
190 KB
diff --git a/‎ves_seg/sample/ng.py‎
Lines changed: 1 addition & 1 deletion b/‎ves_seg/sample/ng.py‎
Lines changed: 1 addition & 1 deletion
@@ -27,7 +27,7 @@ def screenshot(path='temp.png', save=True, show=True, size=None):
 vol_name = '7-13'
 
 clahe = readvol(os.path.join(data_dir, vol_name+"_clahe.h5"))
-mask = readvol(os.path.join(data_dir, vol_name+"_mask.h5"))
+mask = readvol(os.path.join(data_dir, vol_name+"_mask.h5")).astype(np.uint8)
 gt = readvol(os.path.join(data_dir, vol_name+"_ves.h5"))
 pred = readvol(os.path.join(data_dir, vol_name+"_pred.h5"))