masking.py 文件源码

def mask_to_dist(mask):
  """Returns distance transform on a mask

  Arguments:
    mask (array): the mask for which to calculate distance transform

  Returns:
    array: distance transform of the mask
  """
  return nd.distance_transform_edt(mask == 0)