使用保存好的Keras猫狗分类模型

# 使用训练好的模型
from keras.preprocessing import image
import numpy as np
from keras import models
categories = ['Dog', 'Cat']
model = models.load_model('~mypath/cats_and_dogs_small_2.h5')
imgfile = '~my/imgs/cat2.jpg'
img =  image.load_img(imgfile, target_size=(150, 150, 3))
img_data = image.img_to_array(img)
img_data = np.expand_dims(img_data, axis=0)
img_data /= 255

prediction = model.predict(img_data, verbose=1)
print(prediction)
print(categories[1 if prediction[0][0] >= 0.5 else 0])

1/1 [==============================] - 1s 1s/step
[[0.02616626]]
Dog

学了几天keras，觉得还不错，只是学得有点慢。今晚第一次使用训练好的模型来测试识别猫狗图片，识别率还可以。

文件名是cat2，其实里面是狗狗，结果没有错。