在鼠标交互那篇博客中,写了一段用鼠标交互提取图像hsv值的代码:
这一篇对代码进行了一点点改进,从图像改为调用摄像头的视频流进行hsv的提取;
一.完整代码
import cv2
moving = False
upper_H, upper_S, upper_V = 0, 0, 0
lower_H, lower_S, lower_V = 255, 255, 255
def get_HSV(event, x, y, flags, param): # 鼠标回调函数
global upper_H, upper_S, upper_V # 引用全局变量
global lower_H, lower_S, lower_V, moving
if event == cv2.EVENT_LBUTTONDOWN: # 左键按下事件
moving = True
[upper_H, upper_S, upper_V] = frame[y, x]
[lower_H, lower_S, lower_V] = frame[y, x]
elif event == cv2.EVENT_MOUSEMOVE: # 鼠标移动事件
if flags == cv2.EVENT_FLAG_LBUTTON and moving == True: # 左键按下状态
upper_H = frame[y, x, 2] if frame[y, x, 2] > upper_H else upper_H
lower_H = frame[y, x, 2] if frame[y, x, 2] < lower_V else lower_H
upper_S = frame[y, x, 2] if frame[y, x, 2] > upper_S else upper_S
lower_S = frame[y, x, 2] if frame[y, x, 2] < upper_S else upper_S
upper_V = frame[y, x, 2] if frame[y, x, 2] > upper_V else upper_V
lower_V = frame[y, x, 2] if frame[y, x, 2] < lower_V else lower_V
elif event == cv2.EVENT_LBUTTONUP:
moving = False
upper = [upper_H, upper_S, upper_V]
lower = [lower_H, lower_S, lower_V]
print("upper:", upper, "\nlower:", lower)
# 创建一个视频流对象
cap = cv2.VideoCapture(0) # 0表示默认摄像头,如果有多个摄像头可以尝试不同的编号
while True:
# 读取摄像头捕捉到的图像帧
ret, frame = cap.read()
# 如果成功读取到帧
if ret:
# 在窗口中显示图像帧
cv2.imshow('Video', frame)
# 将图像帧从BGR转换为HSV
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
# 注册鼠标回调函数
cv2.setMouseCallback('Video', get_HSV)
# 如果按下q键,则退出程序
if cv2.waitKey(1) == ord('q'): # 按q退出程序
break
# 按下'q'键退出循环
if cv2.waitKey(1) & 0xFF == ord('q'):
break
# 释放视频流对象
cap.release()
# 销毁所有窗口
cv2.destroyAllWindows()