OpenCV  3.0.0-rc1
Open Source Computer Vision
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Groups Pages
Meanshift and Camshift

Goal

In this chapter,

Meanshift

The intuition behind the meanshift is simple. Consider you have a set of points. (It can be a pixel distribution like histogram backprojection). You are given a small window ( may be a circle) and you have to move that window to the area of maximum pixel density (or maximum number of points). It is illustrated in the simple image given below:

meanshift_basics.jpg
image

The initial window is shown in blue circle with the name "C1". Its original center is marked in blue rectangle, named "C1_o". But if you find the centroid of the points inside that window, you will get the point "C1_r" (marked in small blue circle) which is the real centroid of window. Surely they don't match. So move your window such that circle of the new window matches with previous centroid. Again find the new centroid. Most probably, it won't match. So move it again, and continue the iterations such that center of window and its centroid falls on the same location (or with a small desired error). So finally what you obtain is a window with maximum pixel distribution. It is marked with green circle, named "C2". As you can see in image, it has maximum number of points. The whole process is demonstrated on a static image below:

meanshift_face.gif
image

So we normally pass the histogram backprojected image and initial target location. When the object moves, obviously the movement is reflected in histogram backprojected image. As a result, meanshift algorithm moves our window to the new location with maximum density.

Meanshift in OpenCV

To use meanshift in OpenCV, first we need to setup the target, find its histogram so that we can backproject the target on each frame for calculation of meanshift. We also need to provide initial location of window. For histogram, only Hue is considered here. Also, to avoid false values due to low light, low light values are discarded using cv2.inRange() function.

1 import numpy as np
2 import cv2
3 
4 cap = cv2.VideoCapture('slow.flv')
5 
6 # take first frame of the video
7 ret,frame = cap.read()
8 
9 # setup initial location of window
10 r,h,c,w = 250,90,400,125 # simply hardcoded the values
11 track_window = (c,r,w,h)
12 
13 # set up the ROI for tracking
14 roi = frame[r:r+h, c:c+w]
15 hsv_roi = cv2.cvtColor(roi, cv2.COLOR_BGR2HSV)
16 mask = cv2.inRange(hsv_roi, np.array((0., 60.,32.)), np.array((180.,255.,255.)))
17 roi_hist = cv2.calcHist([hsv_roi],[0],mask,[180],[0,180])
18 cv2.normalize(roi_hist,roi_hist,0,255,cv2.NORM_MINMAX)
19 
20 # Setup the termination criteria, either 10 iteration or move by atleast 1 pt
21 term_crit = ( cv2.TERM_CRITERIA_EPS | cv2.TERM_CRITERIA_COUNT, 10, 1 )
22 
23 while(1):
24  ret ,frame = cap.read()
25 
26  if ret == True:
27  hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
28  dst = cv2.calcBackProject([hsv],[0],roi_hist,[0,180],1)
29 
30  # apply meanshift to get the new location
31  ret, track_window = cv2.meanShift(dst, track_window, term_crit)
32 
33  # Draw it on image
34  x,y,w,h = track_window
35  img2 = cv2.rectangle(frame, (x,y), (x+w,y+h), 255,2)
36  cv2.imshow('img2',img2)
37 
38  k = cv2.waitKey(60) & 0xff
39  if k == 27:
40  break
41  else:
42  cv2.imwrite(chr(k)+".jpg",img2)
43 
44  else:
45  break
46 
47 cv2.destroyAllWindows()
48 cap.release()

Three frames in a video I used is given below:

meanshift_result.jpg
image

Camshift

Did you closely watch the last result? There is a problem. Our window always has the same size when car is farther away and it is very close to camera. That is not good. We need to adapt the window size with size and rotation of the target. Once again, the solution came from "OpenCV Labs" and it is called CAMshift (Continuously Adaptive Meanshift) published by Gary Bradsky in his paper "Computer Vision Face Tracking for Use in a Perceptual User Interface" in 1988.

It applies meanshift first. Once meanshift converges, it updates the size of the window as, \(s = 2 \times \sqrt{\frac{M_{00}}{256}}\). It also calculates the orientation of best fitting ellipse to it. Again it applies the meanshift with new scaled search window and previous window location. The process is continued until required accuracy is met.

camshift_face.gif
image

Camshift in OpenCV

It is almost same as meanshift, but it returns a rotated rectangle (that is our result) and box parameters (used to be passed as search window in next iteration). See the code below:

1 import numpy as np
2 import cv2
3 
4 cap = cv2.VideoCapture('slow.flv')
5 
6 # take first frame of the video
7 ret,frame = cap.read()
8 
9 # setup initial location of window
10 r,h,c,w = 250,90,400,125 # simply hardcoded the values
11 track_window = (c,r,w,h)
12 
13 # set up the ROI for tracking
14 roi = frame[r:r+h, c:c+w]
15 hsv_roi = cv2.cvtColor(roi, cv2.COLOR_BGR2HSV)
16 mask = cv2.inRange(hsv_roi, np.array((0., 60.,32.)), np.array((180.,255.,255.)))
17 roi_hist = cv2.calcHist([hsv_roi],[0],mask,[180],[0,180])
18 cv2.normalize(roi_hist,roi_hist,0,255,cv2.NORM_MINMAX)
19 
20 # Setup the termination criteria, either 10 iteration or move by atleast 1 pt
21 term_crit = ( cv2.TERM_CRITERIA_EPS | cv2.TERM_CRITERIA_COUNT, 10, 1 )
22 
23 while(1):
24  ret ,frame = cap.read()
25 
26  if ret == True:
27  hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
28  dst = cv2.calcBackProject([hsv],[0],roi_hist,[0,180],1)
29 
30  # apply meanshift to get the new location
31  ret, track_window = cv2.CamShift(dst, track_window, term_crit)
32 
33  # Draw it on image
34  pts = cv2.boxPoints(ret)
35  pts = np.int0(pts)
36  img2 = cv2.polylines(frame,[pts],True, 255,2)
37  cv2.imshow('img2',img2)
38 
39  k = cv2.waitKey(60) & 0xff
40  if k == 27:
41  break
42  else:
43  cv2.imwrite(chr(k)+".jpg",img2)
44 
45  else:
46  break
47 
48 cv2.destroyAllWindows()
49 cap.release()

Three frames of the result is shown below:

camshift_result.jpg
image

Additional Resources

  1. French Wikipedia page on Camshift. (The two animations are taken from here)
  2. Bradski, G.R., "Real time face and object tracking as a component of a perceptual user interface," Applications of Computer Vision, 1998. WACV '98. Proceedings., Fourth IEEE Workshop on , vol., no., pp.214,219, 19-21 Oct 1998

Exercises

  1. OpenCV comes with a Python sample on interactive demo of camshift. Use it, hack it, understand it.