tfa.image.euclidean_dist_transform

Applies euclidean distance transform(s) to the image(s).

tfa.image.euclidean_dist_transform(
    images: tfa.types.TensorLike,
    dtype: Type[tf.dtypes.DType] = tf.float32,
    name: Optional[str] = None
) -> tf.Tensor

Used in the notebooks

Used in the tutorials
TensorFlow Addons Image: Operations

Based on Distance Transforms of Sampled Functions.

Args
images	A tensor of shape (num_images, num_rows, num_columns, num_channels) (NHWC), or (num_rows, num_columns, num_channels) (HWC) or (num_rows, num_columns) (HW).
dtype	tf.dtypes.DType of the output tensor.
name	The name of the op.

Returns
Image(s) with the type dtype and same shape as images, with the transform applied. If a tensor of all ones is given as input, the output tensor will be filled with the max value of the dtype.

Raises
TypeError	If image is not tf.uint8, or dtype is not floating point.