Функция save_img

Функция save_img сохраняет изображение, представленное в виде тензора или массива, в файл на диске. Первым параметром функция принимает путь к файлу, вторым - данные изображения. Третьим параметром можно передать масштаб значений, четвертым - формат данных изображения.

Синтаксис

tf.keras.utils.save_img(path, x, [scale], [data_format])

Пример

Давайте создадим тензор изображения и сохраним его в файл:

import tensorflow as tf

t = tf.constant([[[255, 0, 0], [0, 255, 0]], [[0, 0, 255], [255, 255, 0]]], dtype=tf.uint8)
tf.keras.utils.save_img('image.png', t)
print("image saved")

Результат выполнения кода:

"image saved"

Пример

Давайте сохраним изображение с указанием масштаба значений:

import tensorflow as tf

t = tf.constant([[[0.5, 0.2, 0.1], [0.3, 0.8, 0.4]], [[0.9, 0.6, 0.7], [0.1, 0.5, 0.2]]], dtype=tf.float32)
tf.keras.utils.save_img('image_scaled.png', t, scale=True)
print("image with scale saved")

Результат выполнения кода:

"image with scale saved"

Пример

Давайте сохраним изображение с указанием формата данных:

import tensorflow as tf

t = tf.constant([[[255, 0, 0], [0, 255, 0]], [[0, 0, 255], [255, 255, 0]]], dtype=tf.uint8)
tf.keras.utils.save_img('image_channels_first.png', t, data_format='channels_first')
print("image with data format saved")

Результат выполнения кода:

"image with data format saved"

Смотрите также

функцию load_img,
которая загружает изображение из файла
функцию img_to_array,
которая преобразует изображение в массив
функцию array_to_img,
которая преобразует массив в изображение
функцию encode_png,
которая кодирует изображение в формат PNG