Перейти к содержимому
Compvision.ru
OmgZomg

Wrong array shape

Recommended Posts

Всем доброго дня!

Загружаю изображение в обученную модель

from keras.models import load_model
import numpy as np


from keras.preprocessing import image
model = load_model('my_model33-16.h5')


img_path='/home/alexander/s_gray/3_39095_31718.png'
img = image.load_img(img_path)
img_tensor = image.img_to_array(img)                  
img_tensor = np.expand_dims(img_tensor, axis=0)        

 

Выдает ошибку

ValueError: Error when checking input: expected img to have shape (128, 128, 1) but got   array with shape (128, 128, 3)

Изображения уже переведены в grayscale

import os
from PIL import Image
files = os.listdir('/home/alexander/s2/')
for f in files:
    img=Image.open('/home/alexander/s2/'+f)
    gray = img.convert('L')
    im_resized = gray.resize((128, 128))
    im_resized.save('/home/alexander/s_gray/'+f)

Что не так?)

Поделиться сообщением


Ссылка на сообщение
Поделиться на других сайтах

Так это, однослойное требует :) 
Проверьте shape загруженного изображения, если слоев 3, просто отцепите один, в сером они все одинаковые.

  • Thanks 1

Поделиться сообщением


Ссылка на сообщение
Поделиться на других сайтах

Создайте учётную запись или войдите для комментирования

Вы должны быть пользователем, чтобы оставить комментарий

Создать учётную запись

Зарегистрируйтесь для создания учётной записи. Это просто!

Зарегистрировать учётную запись

Войти

Уже зарегистрированы? Войдите здесь.

Войти сейчас


  • Сейчас на странице   0 пользователей

    Нет пользователей, просматривающих эту страницу

×