调用深度学习训练的模型利用opencv实现实时预测

from keras.models import load_model
from keras.preprocessing.image import img_to_array, load_img
import numpy as np
model = load_model('cats_and_dogs_1.h5')
classes=['cat','dog']
import cv2
cap = cv2.VideoCapture(0)
while (1):
    ret, frame = cap.read()  # ret:True/False,代表有没有读到图片  frame:当前截取一帧的图片
    cv2.imshow("capture", frame)#展示图像
    if(cv2.waitKey(1) & 0xFF) == ord('s'):#按下s
        img = cv2.resize(frame, (150, 150))#图片大小改为（150，150）
        x = img_to_array(img) / 255.0#数据处理
        x = np.expand_dims(x, axis=0)
        result = model.predict(x)#调用模型预测
        y = (0.5, result[0][0])
        ind=np.argmax(y)
        print('this is a ', classes[ind])
    if (cv2.waitKey(1) & 0xFF) == ord('q'):
        break
cap.release()
cv2.destroyAllWindows()

以最简单的猫狗二分类为例，