tf.image.sobel_edges

Возвращает тензор, содержащий карты границ Собеля.

tf.image.sobel_edges(
    image
)

Пример использования:

Для общего использования image загружался бы из файла, как показано ниже:

image_bytes = tf.io.read_file(path_to_image_file)
image = tf.image.decode_image(image_bytes)
image = tf.cast(image, tf.float32)
image = tf.expand_dims(image, 0)

Но в демонстрационных целях мы используем случайно сгенерированные значения для image:

image = tf.random.uniform(
  maxval=255, shape=[1, 28, 28, 3], dtype=tf.float32)
sobel = tf.image.sobel_edges(image)
sobel_y = np.asarray(sobel[0, :, :, :, 0]) # sobel in y-direction
sobel_x = np.asarray(sobel[0, :, :, :, 1]) # sobel in x-direction

Для отображения результатов Собеля можно использовать модуль Image Module PIL:

# Display edge maps for the first channel (at index 0)
Image.fromarray(sobel_y[..., 0] / 4 + 0.5).show()
Image.fromarray(sobel_x[..., 0] / 4 + 0.5).show()