如何使用python输入图像以便在keras中进行类的预测

时间:2017-10-15 03:53:48

标签: python-3.x machine-learning keras computer-vision

我正在尝试预测单个图像的结果,但它会给出不相关的结果。我已经在cifar 10数据集上训练了模型 我用keras和tensorflow来训练这个模型。我想我提供的输入的大小不正确。

以下是培训代码的要点:https://github.com/09rohanchopra/cifar10/blob/master/cifar10-simple-cnn.ipynb 用于预测simgle图像的代码

from keras.preprocessing import image
from keras.models import load_model
from scipy.misc import imread,imresize
import numpy as np
model=load_model('augmented_best_model.h5')
im=imread('1.jpg')
im=im/255
im=im.resize(im,(32,32))
pr=model.predict(im.reshape(-1,3,32,32))

2 个答案:

答案 0 :(得分:1)

imread现已弃用。更好的选择是使用keras函数。

Item var = (Item) pq.delMax()

答案 1 :(得分:0)

我得到了答案

x=imread('test/2.jpg',mode='RGB')
x=imresize(x,(32,32))
x=np.invert(x)
x=x.reshape(-1,32,32,3)