Изменение формы изображений для модели Conv2D - PullRequest
Новая
       18

Изменение формы изображений для модели Conv2D

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

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

 ValueError: cannot reshape array of size 25000 into shape (50,50,3) ----- BGR

Однако он отлично работает, когда я изменение формы изображений GrayScale. 

X=np.array(x).reshape(-1,50,50,1)

Вот как я загружаю и изменяю размер изображений: 

training_data=[]
labels=[]
for i in os.listdir():
    label=i.split('.')[0]
    if label=='cat':
       labels.append(0)
    else:
       labels.append(1)
    
    image=cv2.imread(i)
    image_array=cv2.resize(image,(50,50))
    training_data.append([image,labels])
...