Skip to content

Commit dd4b9a3

Browse files
committed
Correct typecast from bash call
1 parent e4f73a4 commit dd4b9a3

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

flamingo_tools/segmentation/unet_prediction.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -302,6 +302,10 @@ def distance_watershed_implementation(
302302
else:
303303
input_ = input_path
304304

305+
print(f"Using center distance threshold: {center_distance_threshold}")
306+
print(f"Using boundary distance threshold: {boundary_distance_threshold}")
307+
print(f"Using distance smoothing: {distance_smoothing}")
308+
305309
# Limit the number of cores for parallelization.
306310
n_threads = min(16, mp.cpu_count())
307311

@@ -617,6 +621,9 @@ def run_unet_segmentation_slurm(
617621
This may help to reduce border artifacts. If set to 0 (the default) smoothing is not applied.
618622
"""
619623
min_size = int(min_size)
624+
center_distance_threshold = float(center_distance_threshold)
625+
boundary_distance_threshold = float(boundary_distance_threshold)
626+
distance_smoothing = float(distance_smoothing)
620627
pmap_out = os.path.join(output_folder, "predictions.zarr")
621628
distance_watershed_implementation(pmap_out, output_folder, center_distance_threshold=center_distance_threshold,
622629
boundary_distance_threshold=boundary_distance_threshold,

0 commit comments

Comments
 (0)