快捷導(dǎo)航

在OpenCV里使用Camshift算法的實(shí)現(xiàn)

更新時(shí)間：2019年11月22日 09:28:46 作者：caimouse

這篇文章主要介紹了在OpenCV里使用Camshift算法的實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

前面學(xué)習(xí)過(guò)Meanshift算法，在觀察這個(gè)結(jié)果標(biāo)記時(shí)，會(huì)發(fā)現(xiàn)有這樣一個(gè)問題，如下圖：

汽車比較遠(yuǎn)時(shí)，用一個(gè)很小的窗口就可以把它框住，這是符合近大遠(yuǎn)小的投影原理，當(dāng)比較近的時(shí)候如下：

相同大小的窗口已經(jīng)不能包圍它了，那么這樣跟蹤目標(biāo)對(duì)象就成為了一個(gè)問題，怎么樣來(lái)更改它呢？那么就是Camshift (Continuously Adaptive Meanshift)算法引入的原因了。同時(shí)還會(huì)有一個(gè)問題，怎么樣判斷物體旋轉(zhuǎn)的方向，這個(gè)算法也會(huì)解決這樣的問題。這個(gè)算法發(fā)表在1998年的論文《Computer Vision Face Tracking for Use in a Perceptual User Interface》里。

這個(gè)算法，首先應(yīng)用meanshift找到最大密度，然后再更新窗口的大小，接著計(jì)算最適合外包橢圓；如果不合適又進(jìn)入一輪迭代過(guò)程。直滿足meanshift的條件，并且窗口大小也合適為止。

Camshift函數(shù)返回兩個(gè)值，第一個(gè)值ret是一個(gè)旋轉(zhuǎn)的窗口，第二個(gè)值是窗口搜索位置給下一次搜索使用的。例子如下：

#python 3.7.4,opencv4.1
#蔡軍生 https://blog.csdn.net/caimouse/article/details/51749579
#
import numpy as np
import cv2
from matplotlib import pyplot as plt
 
capture = cv2.VideoCapture(1)
if not capture.isOpened:
  print('Unable to open: ')
  exit(0)
  
#獲取第一幀圖片
ret,frame = capture.read()
 
#設(shè)置目標(biāo)窗口
#讀取文件
find = cv2.imread('luohu1.png')
h,w = find.shape[:2]
roi = find[10: 120, 10: 120]
x = 10
y = 10
width = 120 - x
height = 120 - y
track_window = (x, y, w, h)
print(track_window)
#跟蹤目標(biāo)
hsv_roi = cv2.cvtColor(roi, cv2.COLOR_BGR2HSV)
mask = cv2.inRange(hsv_roi, np.array((0., 60.,32.)), np.array((180.,255.,255.)))
roi_hist = cv2.calcHist([hsv_roi],[0],mask,[180],[0,180]) #計(jì)算直方圖
cv2.normalize(roi_hist,roi_hist,0,255,cv2.NORM_MINMAX)
 
#設(shè)置迭代條件,每10移動(dòng)一點(diǎn)
term_crit = ( cv2.TERM_CRITERIA_EPS | cv2.TERM_CRITERIA_COUNT, 10, 1 )
 
while(1):
  ret, frame = capture.read()
 
  if ret == True:
    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
    dst = cv2.calcBackProject([hsv],[0],roi_hist,[0,180],1)#反向投影
 
    #使用 meanshift獲得新位置
    ret, track_window = cv2.CamShift(dst, track_window, term_crit)
 
    #顯示標(biāo)記
    pts = cv2.boxPoints(ret)
    pts = np.int0(pts)
    img2 = cv2.polylines(frame,[pts],True, (255,0,0),2)
    cv2.imshow('img2',img2)
 
    cv2.imshow("dst", dst) 
    cv2.imshow("roi", roi)
    keyboard = cv2.waitKey(1)
    if keyboard == ord('q') or keyboard == ord('Q'):
      break
  else:
    break
 
  
capture.release()
cv2.destroyAllWindows()

結(jié)果輸出如下：