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

我不知道为什么它会给我这种输出

我怎样才能做到这一点？

提前致谢

标签： pythonmachine-learningkeras

Answer 1

load_img适用于 WH 元组。你得到 (28, 28, 1)(Width, Height, Channels)。如果要将其转换为 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)