![]() |
Applies euclidean distance transform(s) to the image(s).
Aliases:
tfa.image.euclidean_dist_transform(
images,
dtype=tf.float32,
name=None
)
Args:
images
: A tensor of shape (num_images, num_rows, num_columns, 1) (NHWC), or (num_rows, num_columns, 1) (HWC) or (num_rows, num_columns) (HW).dtype
: 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
: Ifimage
is not tf.uint8, ordtype
is not floating point.ValueError
: Ifimage
more than one channel, orimage
is not of rank between 2 and 4.