我想将我的图像调整为(1, 28, 28)的尺寸

我的代码如下

from keras.preprocessing.image import load_imgfrom keras.preprocessing import imageimg1 = 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)

我不知道为什么会得到这种输出

我该怎么做呢？

提前感谢

回答：

load_img 函数使用宽高（WH）元组。你得到的是（28, 28, 1）（宽度，长度，通道）。如果你想将其转换为通道-宽度-高度（CWH）格式，可以使用 numpy.swapaxes。

import numpy as npr = 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)