层序的输入0与层不兼容

Question

I created a model and then loaded it in another script and try to perform a prediction from it however I can not understand why the shape being passed to the function is incorrect.

This is how the model is created:

batch_size = 1232
epochs = 5
IMG_HEIGHT = 400
IMG_WIDTH = 400

model1 = np.load("training_data.npy", allow_pickle=True)
model2 = np.load("training_data_1.npy", allow_pickle=True)

data = np.asarray(np.concatenate((model1, model2), axis=0))  # 1232
train_data = data[:-100]
X_train = np.asarray(np.array([i[0] for i in train_data]))
Y_train = np.asarray([i[1] for i in train_data])

validation_data = data[-100:]

X_val = np.asarray(np.array([i[0] for i in validation_data]))
Y_val = np.asarray([i[1] for i in validation_data])

model = Sequential([
    Conv2D(16, 3, padding='same', activation='relu',
           input_shape=(IMG_HEIGHT, IMG_WIDTH, 3)),
    MaxPooling2D(),
    Conv2D(32, 3, padding='same', activation='relu'),
    MaxPooling2D(),
    Conv2D(64, 3, padding='same', activation='relu'),
    MaxPooling2D(),
    Flatten(),
    Dense(512, activation='relu'),
    Dense(1)
])

model.compile(optimizer='adam',
              loss=tf.keras.losses.BinaryCrossentropy(from_logits=True),
              metrics=['accuracy'])

history = model.fit(X_train, Y_train, steps_per_epoch=batch_size, epochs=epochs,
                    validation_data=(X_val, Y_val), validation_steps=batch_size)

model.save("test")

And this is how I'm trying to make a prediction:

batch_size = 1232
epochs = 5
IMG_HEIGHT = 400
IMG_WIDTH = 400

model = tf.keras.models.load_model('test')

test_1 = cv2.imread('./Data/Images/test_no.jpg')
test_1 = cv2.resize(test_1, (IMG_HEIGHT, IMG_WIDTH))


prediction = model.predict([test_1])[0]
print(prediction)

When printing the shape of the test image the output is: (400, 400, 3)

I also tried using the numpy operation reshape when passing the test image to predict. However the error is always:

ValueError: Input 0 of layer sequential is incompatible with the layer: expected ndim=4, found ndim=3. Full shape received: [None, 400, 3]

Answer 1

Add extra dimension to your input as [n_items,400,400,3]

import tensorflow as tf
X_train = tf.expand_dims(X_train, axis =-1)