LEC-5422-Public/pyTrainSeg
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'merge' into 'master'

Browse Source 
Merge

See merge request fische_r/pytrainseg!1
This commit is contained in:
fische_r
2023-09-19 16:16:47 +02:00
parent 5d8c5d3a7a c6952ebb6a
commit 7e34e0a5fa
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
training_functions.py
+7 -5
View File
@@ -173,21 +173,21 @@ class train_segmentation:
if not c1=='time':
if c2 == 'x':
im = stage1.sel( x = p2).data #feature = 'original',
im = stage1.sel( x = p2)#feature = 'original',
# feat_stack = stage1feat.sel(x = p2).data
imfirst = None
elif c2 == 'y':
im = stage1.sel( y = p2).data
im = stage1.sel( y = p2)
# feat_stack = stage1feat.sel(y = p2).data
imfirst = None
elif c2 == 'z':
im = stage1.sel( z = p2).data
im = stage1.sel( z = p2)
# feat_stack = stage1feat.sel(z = p2).data
imfirst = None
elif c2 == 'time':
im = stage1.sel(time = p2).data
im = stage1.sel(time = p2)
# feat_stack = stage1feat.sel(time = p2).data
imfirst = stage1.sel(time = 0).data
imfirst = stage1.sel(time = 0)
# if self.lazy:
# # get the reference images directly as numpy array
@@ -205,11 +205,13 @@ class train_segmentation:
fut = fut.result()
fut = fut.compute()
im = fut.data
self.client.restart()
if imfirst is not None and type(imfirst) is not np.ndarray:
fut = self.client.scatter(imfirst)
fut = fut.result()
fut = fut.compute()
imfirst = fut.data
self.client.restart()
im8 = im-im.min()