持续调用摄像头,按q退出
代码如下:
import cv2
import time
import numpy as np
import tensorflow as tf
from PIL import Image
with tf.Session() as sess:
video = cv2.VideoCapture(0)
# VideoCapture()中参数是0,表示打开笔记本的内置摄像头,参数是视频文件路径则打开视频
while True:
return_val, frame = video.read()
# read()按帧读取视频,return_val,frame是获video.read()方法的两个返回值。
# 其中return_val是布尔值,如果读取帧是正确的则返回True,如果文件读取到结尾,它的返回值就为False。
# frame就是每一帧的图像,是个三维矩阵。
if return_val:
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) # BGR转RGB
image = Image.fromarray(frame) # 实现array到image的转换
else:
raise ValueError("No image!")
prev_time = time.time() # 记录当前时间
{
# 运行的内容,或对图像的作用
}
curr_time = time.time() # 记录当前时间
exec_time = curr_time - prev_time # 执行{}中内容的用时
result = np.asarray(image)
info = "time: %.2f ms" % (1000 * exec_time)
cv2.putText(result, text=info, org=(20, 20), fontFace=cv2.FONT_HERSHEY_SIMPLEX,
fontScale=1, color=(255, 0, 0), thickness=2)
cv2.namedWindow("result", cv2.WINDOW_AUTOSIZE)
result = cv2.cvtColor(result, cv2.COLOR_RGB2BGR)
cv2.imshow("result", result)
if cv2.waitKey(1) & 0xFF == ord('q'): break