Для проверки типа любого объекта вы можете использовать type()
print( type(train_images) )
Многие модули (не только для Data Science) имеют функцию для загрузки изображения и непосредственного создания массива, либо он может бытьлегко конвертируется с помощью np.array()
.
import numpy as np
import matplotlib.pyplot as plt
filename = 'image.jpg'
import matplotlib.pyplot
img = matplotlib.pyplot.imread(filename)
print(type(img), img.shape)
plt.imshow(img)
plt.show()
Возможно, потребуется преобразовать цвета BGR в RGB
import cv2
img = cv2.imread(filename)
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
print(type(img), img.shape)
plt.imshow(img)
plt.show()
import imageio
img = np.array(imageio.imread(filename))
img = np.array(img)
print(type(img), img.shape)
plt.imshow(img)
plt.show()
import PIL.Image
img = PIL.Image.open(filename)
img = np.array(img)
print(type(img), img.shape)
plt.imshow(img)
plt.show()
Требуется повернуть - поменять оси
import pygame
img = pygame.image.load(filename)
img = pygame.surfarray.array3d(img)
img = img.swapaxes(0, 1)
print(type(img), img.shape)
plt.imshow(img)
plt.show()
import skimage
img = skimage.io.imread(filename)
print(type(img), img.shape)
plt.imshow(img)
plt.show()
import scipy.misc
img = scipy.misc.imread(filename)
print('scipy:', type(img), img.shape)
plt.imshow(img)
plt.show()
Загружается изображение, но отображается предупреждение:
DeprecationWarning: `imread` is deprecated!
`imread` is deprecated in SciPy 1.0.0, and will be removed in 1.2.0.
Use ``imageio.imread`` instead.
Должен быть преобразован из PIL.Image
в numpy.array
from keras.preprocessing.image import load_img
img = load_img(filename)
img = np.array(img)
print('keras:', type(img), img.shape)
plt.imshow(img)
plt.show()