Ошибка при проверке ввода с моделью keras

Question

Я - модель поезда, сохраняю модель на диске и получаю ошибку при использовании модели.

Error when checking input: expected conv2d_input to have 4 dimensions, but got array with shape (150, 150, 3)

Модель поезда:

model = Sequential()

model.add(Conv2D(32, (3, 3), input_shape=(150, 150, 3))

Модель использования:

model = load_model(os.path.join('models', 'myModel.h5'))
model.predict(img) # img - OpenCV image

Answer 1

Используйте это:

import numpy as np
model.predict(img[np.newaxis, :, :, :])