Skip to content

Commit ee8a0ce

Browse files
Fix bug in new membrane post-processing
1 parent efb5851 commit ee8a0ce

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

synapse_net/inference/postprocessing/membranes.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -110,6 +110,7 @@ def segment_membrane_distance_based(
110110
if min_size > 0:
111111
boundary_prediction = label(boundary_prediction, block_shape=(32, 256, 256))
112112
ids, sizes = np.unique(boundary_prediction, return_counts=True)
113+
ids, sizes = ids[1:], sizes[1:]
113114
keep_ids = ids[sizes > min_size]
114115
boundary_prediction = np.isin(boundary_prediction, keep_ids)
115116

0 commit comments

Comments
 (0)