Skip to content

Commit 4862ddb

Browse files
linok-bcMichael Lin
authored andcommitted
fix 7-13 mask again
1 parent 746dabd commit 4862ddb

File tree

2 files changed

+2
-6
lines changed

2 files changed

+2
-6
lines changed

ves_seg/sample/7-13_mask.h5

-23.7 KB
Binary file not shown.

ves_seg/sample/ng.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -27,20 +27,16 @@ def screenshot(path='temp.png', save=True, show=True, size=None):
2727
vol_name = '7-13'
2828

2929
clahe = 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"))
3131
gt = readvol(os.path.join(data_dir, vol_name+"_ves.h5"))
3232
pred = 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
3834
pred = pred * mask
3935

4036
with 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

4642
print(viewer)

0 commit comments

Comments
 (0)