opencv-python学习之入门篇（一）实时追踪蓝色物体

安装opencv-python

pip install opencv-python --user

实时追踪某一蓝色物体

废话不多说，直接上代码：

import numpy as np
import cv2

cap = cv2.VideoCapture(0)

while(True):

#获取每一帧
ret, frame = cap.read()

#转换到HSV
hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)

# 设置HSV阈值
lower_blue = np.array([110, 50, 50])
upper_blue = np.array([130, 255, 255])

# 根据阈值构建掩模
mask = cv2.inRange(hsv, lower_blue, upper_blue)

# 对图像和掩模进行位运算
res = cv2.bitwise_and(frame, frame, mask = mask)

#显示图像
cv2.imshow('frame', frame)
cv2.imshow('mask', mask)
cv2.imshow('res', res)
k = cv2.waitKey(5) & 0xFF
if k == 27:
break

cap.release()
cv2.destroyAllWindows()

当然，还可以查HSV表追踪其他颜色的物体。