在 Keras 中重塑图像数据以满足 CNN 要求

Question

我创建了一个专为识别物体而设计的 CNN。

from keras.preprocessing.image import img_to_array, load_img

img = load_img('newimage.jpg')
x = img_to_array(img)
x = x.reshape( (1,) + x.shape )
scores = model.predict(x, verbose=1)
print(scores)

但是我得到：

expected convolution2d_input_1 to have shape (None, 3, 108, 192) but got array with shape (1, 3, 192, 108)

我的模特：

def create_model():
    model = Sequential()
    model.add(Convolution2D(32, 3, 3, input_shape=(3, img_width, img_height)))
    model.add(Activation('relu'))
    model.add(MaxPooling2D(pool_size=(2, 2)))

    model.add(Convolution2D(32, 3, 3))
    model.add(Activation('relu'))
    model.add(MaxPooling2D(pool_size=(2, 2)))

    model.add(Convolution2D(64, 3, 3))
    model.add(Activation('relu'))
    model.add(MaxPooling2D(pool_size=(2, 2)))

    model.add(Flatten())
    model.add(Dense(64))
    model.add(Activation('relu'))
    model.add(Dropout(0.5))

    model.add(Dense(1))
    model.add(Dense(3, activation='softmax'))

    model.compile(loss='categorical_crossentropy',
                  optimizer='rmsprop',
                  metrics=['accuracy'])
    return model

我查看了相关的答案和文档，但不知道如何重塑数组以匹配预期的内容？

Answer 1

我想问题出在设置图片的宽度和高度上。正如错误所说：

expected convolution2d_input_1 to have shape (None, 3, 108, 192) # expected width = 108 and height = 192 
but got array with shape (1, 3, 192, 108) # width = 192, height = 108

Update: I tested your code with a small change and it worked!

我给出的是刚刚更改的行：

img_width, img_height = 960, 717
model.add(Convolution2D(32, 3, 3, input_shape=(img_height, img_width, 3)))

这是主要变化 - input_shape=(img_height, img_width, 3)

我用来运行此代码的图像是 width = 960 和 height = 717。我已经更新了我之前的答案，因为部分答案是错误的！抱歉。

在 Keras 中重塑图像数据以满足 CNN 要求

Reshaping image data in Keras to match CNN requirements

deep-learning

keras