Keras中加载已训练模型并识别猫狗图片分类的案例

from keras import models

from keras.preprocessing import image

import numpy as np

 

#加载预训练模型

model = models.load_model("cats_and_dogs_small_model.h5")

#准备测试图片各一张

#img_dir = "D:/Documents/vscodedoc/data/kaggle/smalldata/test/dogs/dog.1507.jpg"

img_dir = "D:/Documents/vscodedoc/data/kaggle/smalldata/test/cats/cat.1502.jpg"

#加载图像

img = image.load_img(img_dir, target_size=(150,150))

#转换为张量(150,150,3)

img_data = image.img_to_array(img)

#转换为模型需要的4D张量(1,150,150,3)

img_data = np.expand_dims(img_data, axis=0)

#和训练时一致，将张量数据除以255将值控制在0-1之间

img_data /= 255

#分类

r = model.predict(img_data)

print(r)