一、環境
win10、Python3.6、OpenCV3.x；編譯器：pycharm5.0.3
二、實現目標
根據需要追蹤的物體顏色，設定閾值，在視頻中框選出需要追蹤的物體。
三、實現步驟
1）根據需要追蹤的物體顏色，設定顏色閾值，獲取追蹤物體的掩膜
代碼：generate_threshold.py
# -*- coding : utf-8 -*-
# Author: Tom Yu
import cv2
import numpy as np
cap = cv2.VideoCapture(0)#獲取攝像頭圖像
# img = cv2.imread("timg1.jpg")
# hsv_img = cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
def nothing(x):
pass
def createbars():
"""
實現創建六個滑塊的作用，分別控制H、S、V的最高值與最低值
"""
cv2.createTrackbar("H_l","image",0,180,nothing)
cv2.createTrackbar("H_h","image",0,180,nothing)
cv2.createTrackbar("S_l","image",0,255,nothing)
cv2.createTrackbar("S_h","image",0,255,nothing)
cv2.createTrackbar("V_l","image",0,255,nothing)
cv2.createTrackbar("V_h","image",0,255,nothing)
cv2.namedWindow("image")
createbars()#創建六個滑塊
lower = np.array([0,0,0])#設置初始值
upper = np.array([0,0,0])
while True:
ret,frame = cap.read()
hsv_frame = cv2.cvtColor(frame,cv2.COLOR_BGR2HSV)#將圖片由BGR顏色空間轉化成HSV空間，HSV可以更好地分割顏色圖形
lower[0]=cv2.getTrackbarPos("H_l","image")#獲取"H_l"滑塊的實時值
upper[0]=cv2.getTrackbarPos("H_h","image")#獲取"H_h"滑塊的實時值
lower[1]=cv2.getTrackbarPos("S_l","image")
upper[1]=cv2.getTrackbarPos("S_h","image")
lower[2]=cv2.getTrackbarPos("V_l","image")
upper[2]=cv2.getTrackbarPos("V_h","image")
mask = cv2.inRange(hsv_frame,lower,upper)#cv2.inrange()函數通過設定的最低、最高閾值獲得圖像的掩膜
cv2.imshow("img",frame)
cv2.imshow("mask",mask)
if cv2.waitKey(1)&0xff == 27:
break
cv2.destroyAllWindows()
實現效果：獲取需要追蹤的物體顏色閾值
2）根據獲取到的閾值，設定閾值范圍，在視頻中追蹤特定顏色的物體并用框選框出所需追蹤的物體
代碼：tracking_object.py
# -*- coding : utf-8 -*-
# Author: Tom Yu
import cv2
import numpy as np
cap = cv2.VideoCapture(0)#獲取攝像頭視頻
while True:
ret,frame = cap.read()#讀取每一幀圖片
hsv_frame = cv2.cvtColor(frame,cv2.COLOR_BGR2HSV)#將每一幀圖片轉化HSV空間顏色
"""
依據之前的腳本獲取的閾值設置最高值與最低值
"""
lower = np.array([0,104,205])
upper = np.array([15,208,255])
mask = cv2.inRange(hsv_frame,lower,upper)
img,conts,hier = cv2.findContours(mask,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)#找出邊界
cv2.drawContours(frame,conts,-1,(0,255,0),3)#畫出邊框
dst = cv2.bitwise_and(frame,frame,mask=mask)#對每一幀進行位與操作，獲取追蹤圖像的顏色
#cv2.imshow("mask",mask)
#cv2.imshow("dst",dst)
cv2.imshow("frame",frame)
if cv2.waitKey(1)&0xff == 27:
break
cv2.destroyAllWindows()
實現效果：
以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。
更多文章、技術交流、商務合作、聯系博主
微信掃碼或搜索：z360901061
微信掃一掃加我為好友
QQ號聯系： 360901061
您的支持是博主寫作最大的動力，如果您喜歡我的文章，感覺我的文章對您有幫助，請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧，狠狠點擊下面給點支持吧，站長非常感激您！手機微信長按不能支付解決辦法：請將微信支付二維碼保存到相冊，切換到微信，然后點擊微信右上角掃一掃功能，選擇支付二維碼完成支付。
【本文對您有幫助就好】元
