如何解决img = np.reshape(img,[1,224,224,3]) - ValueError: 无法将大小为 200704 的数组重塑为形状 (1,224,224,3)
我创建了一个模型来预测图像的类别。我需要将此图像重塑为特定形状 (1,224,3)
。我已经使用这个模型来预测其他一些图像,它工作得很好。但是,当我使用此图像时,它不起作用。这里有什么问题,我应该修改什么?
我使用此代码读取图像并预测其类别:
from skimage.io import imread
from skimage.transform import resize
import numpy as np
import math
img = imread('/path/to/image.PNG')
img = resize(img,(224,224))
img = img*1./255
img = np.reshape(img,[1,3])
classes = model.predict(img)
并得到这个错误:
ValueError: cannot reshape array of size 200704 into shape (1,3)
编辑:在我看来,这个图像中有四个颜色通道而不是 3 个(224x224x4=200.704;但图像应该只是一个大小为 224x224x3 的数组)。我该如何解决这个问题?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。