如何为机器学习模型调整图像大小

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)。我不知道为什么会得到这种输出，我该怎么做呢？谢谢！

- Jose

---

@Nakor，numpy.swapaxes能否给我这个尺寸？ - Jose

Answer 1

load_img 函数处理 WH 元组，也就是宽度、高度和通道数（Width, Height, Channels），如果你想要将其转换为 CWH 格式，请使用 numpy.swapaxes。

注：WH 和 CWH 分别表示宽度、高度和通道数的顺序。

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)

- Sıddık Açıl