tfaddons (version 0.10.0)

img_euclidean_dist_transform: Euclidean dist transform

Description

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

Usage

img_euclidean_dist_transform(images, dtype = tf$float32, name = NULL)

Arguments

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.

Value

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. ValueError: If `image` more than one channel, or `image` is not of rank between 2 and 4.

Examples

# NOT RUN {
# }
# NOT RUN {
img_path = tf$keras$utils$get_file('tensorflow.png','https://tensorflow.org/images/tf_logo.png')
img_raw = tf$io$read_file(img_path)
img = tf$io$decode_png(img_raw)
img = tf$image$convert_image_dtype(img, tf$float32)
img = tf$image$resize(img, c(500L,500L))
bw_img = 1.0 - tf$image$rgb_to_grayscale(img)
gray = tf$image$convert_image_dtype(bw_img,tf$uint8)
gray = tf$expand_dims(gray, 0L)
eucid = img_euclidean_dist_transform(gray)
eucid = tf$squeeze(eucid, c(0,-1))
# }
# NOT RUN {

# }