2017-10-15 4 views
0

단일 이미지에 대한 결과를 예측하려고하지만 관련성이없는 결과가 나타납니다. https://github.com/09rohanchopra/cifar10/blob/master/cifar10-simple-cnn.ipynb 코드 내가 가진keras : 단일 이미지에서 예측

#
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

왜 이미지를 채널로 먼저 바꿨습니까? 훈련 코드에서 이미지 입력 ​​형태는'input_shape = (IMAGE_SIZE, IMAGE_SIZE, CHANNELS)'입니다. – Matin

+0

또한 im이 정수형의 dtype을 가지고 있다면,'im/255'는 파이썬 2에서 그것을 0으로 만들 것입니다 –

답변

0

simgle 이미지를 예측 : 나는이 모델 여기

를 양성하는 keras 및 tensorflow을 사용했다 cifar 10 세트 에 모델을 훈련 한 훈련 코드의 요점이다 대답

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

imread는 더 이상 사용되지 않습니다. 더 나은 대안은 keras 함수를 사용하는 것입니다.

x = image.load_img('test/2.jpg', target_size=(32,32)) 
x = image.img_to_array(x) 
x = x.reshape((1,) + x.shape) 
x = x/255.