赞
踩
- pip install opencv-python
- pip install tensorflow
load_model
函数来加载模型。cv2.VideoCapture
函数来读取摄像头图像,然后将图像传递给你训练好的模型进行预测。以下是一个示例代码,可以识别出植物是 "玫瑰" 或 "郁金香":
- import cv2
- import tensorflow as tf
- import numpy as np
-
- # 加载模型
- model = tf.keras.models.load_model('my_model.h5')
-
- # 定义分类标签
- labels = ['Rose', 'Tulip']
-
- # 打开摄像头
- cap = cv2.VideoCapture(0)
-
- while True:
- # 读取摄像头图像
- ret, frame = cap.read()
-
- # 调整图像大小
- frame = cv2.resize(frame, (224, 224))
-
- # 归一化图像
- frame = frame / 255.0
-
- # 添加批次维度
- frame = np.expand_dims(frame, axis=0)
-
- # 进行预测
- predictions = model.predict(frame)
-
- # 获取预测结果
- label = labels[np.argmax(predictions)]
-
- # 在图像上绘制标签
- cv2.putText(frame, label, (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
-
- # 显示图像
- cv2.imshow('Plant Recognition', frame)
-
- # 按下 'q' 键退出循环
- if cv2.waitKey(1) & 0xFF == ord('q'):
- break
-
- # 释放资源
- cap.release()
- cv2.destroyAllWindows()
在这个示例代码中,已经训练好的模型,该模型可以识别出 "玫瑰" 和 "郁金香" 两种植物。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。