def distancetransf(image):
if image.dtype=='bool':
return ndimage.distance_transform_edt(1-image.astype(np.int8))
else:
return ndimage.distance_transform_edt(1-image)
微信
公众号
扫码关注公众号