为Conv2D模型重塑图像

时间:2020-08-03 03:11:15

标签: python numpy tensorflow

我正在建立用于二进制分类的分类器,当我尝试将图像重塑为模型的正确输入类型时,我会遇到此错误。

 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])

0 个答案:

没有答案