如何为机器学习模型调整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)