Skip to content

Commit abb0308

Browse files
perf: delete dt sooner in spherical_dilate
1 parent 8f641d2 commit abb0308

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

fastmorph/__init__.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -208,15 +208,15 @@ def spherical_dilate(
208208
"""
209209
if np.issubdtype(labels.dtype, bool):
210210
dt = edt.edt(labels == 0, parallel=parallel, anisotropy=anisotropy)
211-
212-
binary_image = lambda: dt <= radius
211+
binary_image = dt <= radius
212+
del dt
213+
213214
if in_place:
214-
labels |= binary_image()
215+
labels |= binary_image
215216
return labels
216217

217-
binimg = binary_image()
218-
binimg |= labels
219-
return binimg
218+
binary_image |= labels
219+
return binary_image
220220

221221
import scipy.ndimage
222222
dt, indices = scipy.ndimage.distance_transform_edt(

0 commit comments

Comments
 (0)