Функция img_to_array

Функция img_to_array преобразует изображение, загруженное через PIL, в массив NumPy. Первым параметром функция принимает изображение PIL. Вторым параметром можно передать тип данных элементов будущего массива, третьим - порядок следования каналов.

Синтаксис

tf.keras.utils.img_to_array(img, [data_format], [dtype])

Пример

Давайте загрузим изображение с диска и преобразуем его в массив NumPy:

import tensorflow as tf

img = tf.keras.utils.load_img('image.jpg')
res = tf.keras.utils.img_to_array(img)

print(type(res))
print(res.shape)

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

<class 'numpy.ndarray'>
(224, 224, 3)

Пример

Давайте преобразуем изображение в массив с типом float32:

import tensorflow as tf

img = tf.keras.utils.load_img('image.jpg')
res = tf.keras.utils.img_to_array(img, dtype=tf.float32)

print(res.dtype)
print(res.shape)

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

<class 'numpy.float32'>
(224, 224, 3)

Пример

Давайте преобразуем изображение в массив с форматом каналов channels_first:

import tensorflow as tf

img = tf.keras.utils.load_img('image.jpg')
res = tf.keras.utils.img_to_array(img, data_format='channels_first')

print(res.shape)

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

(3, 224, 224)

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

функцию load_img,
которая загружает изображение с диска
функцию array_to_img,
которая преобразует массив NumPy в изображение PIL
функцию save_img,
которая сохраняет изображение на диск
функцию convert_image_dtype,
которая преобразует тип данных изображения