@@ -27,20 +27,16 @@ def screenshot(path='temp.png', save=True, show=True, size=None):
2727vol_name = '7-13'
2828
2929clahe = readvol (os .path .join (data_dir , vol_name + "_clahe.h5" ))
30- mask = readvol (os .path .join (data_dir , vol_name + "_mask.h5" )). astype ( np . uint8 )
30+ mask = readvol (os .path .join (data_dir , vol_name + "_mask.h5" ))
3131gt = readvol (os .path .join (data_dir , vol_name + "_ves.h5" ))
3232pred = readvol (os .path .join (data_dir , vol_name + "_pred.h5" ))
3333
34- mask_orig = mask
35- for i in range (2 ):
36- mask = morphology .binary_dilation (mask , morphology .ball (radius = 2 ))
37- gt = gt * mask
3834pred = pred * mask
3935
4036with viewer .txn () as s :
4137 s .layers .append (name = 'clahe' ,layer = ng_layer (clahe , res = [8 , 8 , 30 ], tt = 'image' ))
4238 s .layers .append (name = 'gt' ,layer = ng_layer (gt , res = [8 , 8 , 30 ], tt = 'segmentation' ))
43- s .layers .append (name = 'mask' ,layer = ng_layer (mask_orig , res = [8 , 8 , 30 ]))
39+ s .layers .append (name = 'mask' ,layer = ng_layer (mask , res = [8 , 8 , 30 ]))
4440 s .layers .append (name = 'pred' ,layer = ng_layer (pred , res = [8 , 8 , 30 ], tt = 'segmentation' ))
4541
4642print (viewer )
0 commit comments