如何调整机器学习模型的img大小答案

【问题标题】：how to resize img for machine learning model如何调整机器学习模型的img大小
【发布时间】：2019-11-15 12:01:50
【问题描述】：

我想将我的图片大小调整为(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)

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

我该怎么做？

提前致谢

【问题讨论】：

@Nakor numpy.swapaxes 能给我这个尺寸吗？

标签： python machine-learning keras

【解决方案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)

【讨论】：