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

adapted vdb creation for sample

Browse Source
This commit is contained in:
Fischer Robert
2023-07-26 12:57:16 +02:00
parent 4741a84d4e
commit 9ef8dcc47f
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
03_nc_to_npy_for_openvdb.py
View File

@@ -62,7 +62,7 @@ class volume_maker:
def clean_binary_image(self, im, clean = True, remove_small=True, minsize = 20, fp_radius = 1, i=0, GPU = True, GPU_avail=GPU_avail): def clean_binary_image(self, im, clean = True, remove_small=True, minsize = 20, fp_radius = 1, i=0, GPU = True, GPU_avail=GPU_avail):
if clean or remove_small: if clean or remove_small:
if GPU and GPU_avail: if GPU and GPU_avail:
gpu_id = i%num_GPU #use gpus 1 through 4, leaving the big A40 (0) alone or i%5 to use all 5 gpu_id = i%4+1 #num_GPU #use gpus 1 through 4, leaving the big A40 (0) alone or i%5 to use all 5
with cp.cuda.Device(gpu_id): with cp.cuda.Device(gpu_id):
im = cp.array(im) im = cp.array(im)
@@ -87,7 +87,7 @@ class volume_maker:
mask = self.data[self.mask_name].sel(timestep = ts).data mask = self.data[self.mask_name].sel(timestep = ts).data
if GPU and GPU_avail: if GPU and GPU_avail:
gpu_id = i%num_GPU #use gpus 1 through 4, leaving the big A40 (0) alone or i%5 to use all 5 gpu_id = i%4+1 #num_GPU #use gpus 1 through 4, leaving the big A40 (0) alone or i%5 to use all 5
with cp.cuda.Device(gpu_id): with cp.cuda.Device(gpu_id):
mask = cp.array(mask) mask = cp.array(mask)
@@ -153,7 +153,7 @@ if __name__ == '__main__':
parser.add_argument('-ph', '--phase', type = int, default = 1, help='which phase to extract, -1 for all') parser.add_argument('-ph', '--phase', type = int, default = 1, help='which phase to extract, -1 for all')
parser.add_argument('-mk', '--mask', type = bool, default = False, help='wheter to use the mask in the segmented data (if available)') parser.add_argument('-mk', '--mask', type = bool, default = False, help='wheter to use the mask in the segmented data (if available)')
parser.add_argument('-mn', '--mask_name', type = str, default = '', help='name of the mask in the segmented data (if available)') parser.add_argument('-mn', '--mask_name', type = str, default = '', help='name of the mask in the segmented data (if available)')
parser.add_argument('-md', '--mask_dilate', type = int, default = 3, help='dilation radius of mask') parser.add_argument('-md', '--mask_dilate', type = int, default = 8, help='dilation radius of mask')
args = parser.parse_args() args = parser.parse_args()