1.如何使用这个模型?我们用模型来对自己的图片分类。
# 图像处理
from skimage.transform import resize
my_image = 'img_test/8.jpg' # 自己图片的路径,可以是相对路径,也可是绝对路径
my_label_y = [1] # 标记一下是否是猫
my_image = np.array(plt.imread(my_image))
plt.imshow(my_image) # 展示图片
num_px = 64
my_image = resize(my_image, (num_px, num_px)) # 重新调整图片大小
my_image = my_image.reshape(1,-1) # 向量,这里已经对里面的值做了归一化,因为上面的resize有一个参数默认可以做归一化
result = cat_model.predict(my_image) # 调用模型
if result > 0.5: # 如果模型预测值大于0.5,我们就认为是猫
print('猫')
else:
print('非猫')