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

fixed bugs for masking

Browse Source
This commit is contained in:
Fischer Robert
2023-07-13 16:52:55 +02:00
parent ed70f0c687
commit 55a540a63e
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
03_nc_to_npy_for_openvdb.py
View File

@@ -54,7 +54,7 @@ class volume_maker:
self.topoutfolder = args.output_path
self.array_name = args.segmented_name
self.ph = args.phase
self.mask = args.masks
self.mask = args.mask
self.mask_name = args.mask_name
@@ -81,6 +81,8 @@ class volume_maker:
if self.mask:
# TODO: check if mask is int-binary 0-1 and adjust if necessary
a,b,c,d,e,f = self.data.attrs['cropping of seg data']
im = im[a:b,c:d,e:f]
mask = self.data[self.mask_name].sel(timestep = ts).data
im = im*mask

6
04_npy_to_openvdb_in_blender_combined.py
View File

@@ -14,9 +14,9 @@ import numpy as np
import pyopenvdb as openvdb
toppath = '/mpc/homes/fische_r/NAS/DASCOELY/processing/04_membrane_ML/5/membrane_npy'
toppath2 = '/mpc/homes/fische_r/NAS/DASCOELY/processing/04_membrane_ML/5/crack_npy'
topoutpath = '/mpc/homes/fische_r/NAS/DASCOELY/processing/04_membrane_ML/5/combined_vdb'
toppath = '/mpc/homes/fische_r/NAS/DASCOELY/processing/04_membrane_ML/5II/membrane_npy'
toppath2 = '/mpc/homes/fische_r/NAS/DASCOELY/processing/04_membrane_ML/5II/crack_npy'
topoutpath = '/mpc/homes/fische_r/NAS/DASCOELY/processing/04_membrane_ML/5II/combined_vdb'
if not os.path.exists(topoutpath):
os.mkdir(topoutpath)