源码:
import cv2 as cv
import numpy as np
# # 使用hsv色彩空间和inRange()API来实现视频的色彩追踪
# def extrace_object_demo():
# capture = cv.VideoCapture("C:/cvtest/0001.mp4")
# while(True):
# ret, frame = capture.read()
# if not ret:
# break
# hsv = cv.cvtColor(frame, cv.COLOR_BGR2HSV)
# lower_hsv = np.array([26, 43, 46])
# upper_hsv = np.array([34, 255, 255])
# dst = cv.inRange(hsv, lowerb=lower_hsv, upperb=upper_hsv)
# cv.imshow("video", frame)
# cv.imshow("extrace_object_demo", dst)
# c = cv.waitKey(40)
# if c == 27:
# break
# 同上,并把追踪到的部分还原为彩色。
def extrace_object_demo():
capture = cv.VideoCapture("C:/cvtest/0001.mp4")
while(True):
ret, frame = capture.read()
if not ret:
break
hsv = cv.cvtColor(frame, cv.COLOR_BGR2HSV)
lower_hsv = np.array([26, 43, 46])
upper_hsv = np.array([34, 255, 255])
mask = cv.inRange(hsv, lowerb=lower_hsv, upperb=upper_hsv)
dst = cv.bitwise_and(frame, frame, mask=mask)
cv.imshow("video", frame)
cv.imshow("mask", dst)
c = cv.waitKey(40)
if c == 27:
break
def color_space_demo(image):
gray = cv.cvtColor(image, cv.COLOR_BGR2GRAY)
cv.imshow("gray", gray)
hsv = cv.cvtColor(image, cv.COLOR_BGR2HSV)
cv.imshow("hsv", hsv)
yuv = cv.cvtColor(image, cv.COLOR_BGR2YUV)
cv.imshow("yuv", yuv)
Ycrcb = cv.cvtColor(image, cv.COLOR_BGR2YCrCb)
cv.imshow("Ycrcb", Ycrcb)
his = cv.cvtColor(image, cv.COLOR_BGR2HLS)
cv.imshow("his", his)
print("-------------Hello Python--------------")
src = cv.imread("C:/cvtest/0000.jpg") # blue, green, red
cv.namedWindow("input image", cv.WINDOW_AUTOSIZE)
cv.imshow("input image", src)
# 通道分离
b, g, r = cv.split(src)
cv.imshow("blue", b)
cv.imshow("green", g)
cv.imshow("red", r)
# 把某个通道的值全弄没
src[:, :, 0] = 0
# 通道合并
src = cv.merge([b, g, r])
cv.imshow("changed image", src)
extrace_object_demo()
cv.waitKey(0)
cv.destroyAllWindows()