Keras 打开图像并使用它们创建批处理

Question

我有以下从 SO 读取图像的功能：

from keras.preprocessing import image as kimage
from keras.applications.vgg16 import preprocess_input

def read_image(path):
    img = kimage.load_img(path, target_size=(224, 224))
    tmp = kimage.img_to_array(img)
    tmp = np.expand_dims(tmp, axis=0)
    tmp = preprocess_input(tmp)
    return tmp

然后我创建了以下数据生成器，它基本上迭代我的路径字符串并调用前面的函数。我想调用读取图像函数并将其输出堆叠到一个可以输入到 keras fit 函数的 numpy 数组中。我正在使用此代码：

batch_holder = np.zeros((batch_size, 224, 244, 3))
for j, row in batch.iterrows():
    batch_holder[j, :] = read_image(row['path'])[0]

但这不起作用，我收到以下我认为无法解释的错误：

文件“train.py”，第 71 行，在 data_generator 中 batch_holder[j, :] = read_image(row['path'])[0] ValueError: could not broadcast input array from shape (224,224,3) into shape (224,244,3)

我做错了什么？

Answer 1

你是不是写错了 244 而不是 224？

batch_holder = np.zeros((batch_size, 224, 244, 3))