Функция load_img

Функция load_img загружает изображение из указанного файла и возвращает объект PIL Image. Первым параметром функция принимает путь к файлу изображения. Вторым параметром можно передать целевой размер изображения. Третьим параметром задается цветовой режим. Четвертым параметром можно указать интерполяцию. Пятым параметром передается коэффициент масштабирования.

Синтаксис

tf.keras.utils.load_img(path, [grayscale], [color_mode], [target_size], [interpolation], [keep_aspect_ratio])

Пример

Давайте загрузим изображение из файла и выведем его тип и размер:

import tensorflow as tf

img = tf.keras.utils.load_img('image.jpg')
print(type(img))
print(img.size)

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

<class 'PIL.JpegImagePlugin.JpegImageFile'>
(640, 480)

Пример

Давайте загрузим изображение с указанием целевого размера:

import tensorflow as tf

img = tf.keras.utils.load_img('image.jpg', target_size=(224, 224))
print(img.size)

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

(224, 224)

Пример

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

import tensorflow as tf

img = tf.keras.utils.load_img('image.jpg', color_mode='grayscale')
print(img.mode)

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

"L"

Пример

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

import tensorflow as tf

img = tf.keras.utils.load_img('image.jpg', target_size=(224, 224), interpolation='bilinear')
print(img.size)

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

(224, 224)

Пример

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

import tensorflow as tf
import numpy as np

img = tf.keras.utils.load_img('image.jpg', target_size=(224, 224))
arr = tf.keras.utils.img_to_array(img)
print(arr.shape)

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

(224, 224, 3)

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

функцию img_to_array,
которая преобразует изображение PIL в массив NumPy
функцию array_to_img,
которая преобразует массив NumPy в изображение PIL
функцию save_img,
которая сохраняет изображение в файл
функцию image_dataset_from_directory,
которая создает датасет из изображений в директории