Как изменить размер IMG для модели машинного обучения

Question

я хочу изменить размер моего изображения на (1, 28, 28)

мой код здесь

from keras.preprocessing.image import load_img
from keras.preprocessing import image

img1 = image.load_img(path=path,grayscale=True,target_size=(28,28))
img1 = image.img_to_array(img1)

print(img1.shape)
## (28, 28, 1)

если target_size=(1,28,28), то это дает мне (1, 28, 1)

Я не знаю, почему это дает мне такой вывод

как я могу это сделать?

Заранее спасибо

Answer 1

load_img работает с кортежами WH. Вы получаете (28, 28, 1) (ширина, высота, каналы). Если вы хотите преобразовать это в формат CWH, используйте numpy.swapaxes .

import numpy as np

r = np.random.rand(3, 3, 1)
r_swapped = np.swapaxes(r, 0, 2)
print(r)
print(r.shape)
print(r_swapped)
print(r_swapped.shape)