Преобразование двухмерной шкалы серого в четырехмерную для вывода модели Keras - PullRequest
0 голосов
/ 05 августа 2020

У меня есть предварительно обученная модель Keras, которую мне нужно использовать для классификации изображения 512x 512, которое изначально находится в формате оттенков серого. Вход в модель Keras должен иметь форму (None, 512, 512, 1). введите описание изображения здесь .

Я выполнил следующий код:

model=load_model('model.h5')
img = Image.open('img.jpg')
img_array = np.array (img)
img_array = img_array/255
model.predict (img_array)

Однако я получаю следующую ошибку:

Ошибка при проверке ввода: ожидалось, что input_1 будет иметь 4 измерения, но получил массив с формой (512, 512)

Я знаю, что мне нужно преобразовать мое изображение в оттенках серого в 4D, чтобы оно соответствовало желаемой форме ввода, однако я не уверен, как это сделать это так, чтобы изображение сохраняло свои первоначальные черты. Как правильно преобразовать изображение в оттенках серого в 4D?

Спасибо.

1 Ответ

0 голосов
/ 05 августа 2020

попробуйте изменить форму массива

img_array = img_array.reshape((1, 512, 512, 1)) 

здесь 1-е и последнее измерение - это размер пакета и каналы соответственно

...