原理如下:
使用opencv进行视频捕捉,然后将RGB图像转为HSV图像,根据颜色范围找出特定颜色,从而实现对这个目标的追踪
代码如下:
# coding: utf-8
import cv2 as cv
import numpy as np
if __name__ == "__main__":
# print all colors
flags = [i for i in dir(cv) if i.startswith("COLOR_")]
print flags
### Object tracking ###
# open device
cap = cv.VideoCapture(0)
# set blue range in HSV mode
lower_blue = np.array([110, 50, 50])
upper_blue = np.array([130, 255, 255])
# start capture and tracking color
while True:
# capture one frame
ret, frame = cap.read()
# do simple proccessing
hsv = cv.cvtColor(frame, cv.COLOR_BGR2HSV) # RGB->HSV
mask = cv.inRange(hsv, lower_blue, upper_blue) # set mask for HSV image
res = cv.bitwise_and(frame, frame, mask=mask) # do bitwise-and mask
# show image
cv.imshow("frame", frame)
cv.imshow("hsv", hsv)
cv.imshow("mask", mask)
cv.imshow("res", res)
# judge exit
if cv.waitKey(5) & 0xff == 27:
break
# release all
cv.destroyAllWindows()
效果如下:
使用手机显示蓝色图片,使用程序进行追踪