I thought it'd be fun to try out some object tracking in OpenCV and as you can see it works quite well. The code is here and the video below shows the actual real-time tracking.I thought it'd be fun to try out some object tracking in OpenCV and as you can see it works quite well. The code is here and the video below shows the actual real-time tracking.
Your code does not work, but mine does
ReplyDeleteimport cv
def find_car(image):
size = cv.GetSize(image)
#prepare memory
car = cv.CreateImage(size, 8, 1)
red = cv.CreateImage(size, 8, 1)
hsv = cv.CreateImage(size, 8, 3)
sat = cv.CreateImage(size, 8, 1)
#split image into hsv, grab the sat
cv.CvtColor(image, hsv, cv.CV_BGR2HSV)
cv.Split(hsv, None, sat, None, None)
#split image into rgb
cv.Split(image, None, None, red, None)
#find the car by looking for red, with high saturation
cv.Threshold(red, red, 128, 255, cv.CV_THRESH_BINARY)
cv.Threshold(sat, sat, 128, 255, cv.CV_THRESH_BINARY)
#AND the two thresholds, finding the car
cv.Mul(red, sat, car)
#remove noise, highlighting the car
cv.Erode(car, car, iterations=5)
cv.Dilate(car, car, iterations=5)
storage = cv.CreateMemStorage(0)
obj = cv.FindContours(car, storage, cv.CV_RETR_CCOMP, cv.CV_CHAIN_APPROX_SIMPLE)
cv.ShowImage('A', car)
if not obj:
return(0, 0, 0, 0)
else:
return cv.BoundingRect(obj)
points = []
capture = cv.CaptureFromCAM(0)
if not capture:
print "Error opening capture device"
sys.exit(1)
cv.SetCaptureProperty(capture, cv.CV_CAP_PROP_FRAME_WIDTH, 640)
cv.SetCaptureProperty(capture, cv.CV_CAP_PROP_FRAME_HEIGHT, 480)
while 1:
original = cv.QueryFrame(capture)
car_rect = find_car(original)
print car_rect
middle = (car_rect[0] + (car_rect[2] / 2), car_rect[1] + (car_rect[3]/2))
if points == []:
points.append(middle)
else:
if abs(points[-1][0] - middle[0]) > 5 and abs(points[-1][1] - middle[1]) > 10:
points.append(middle)
cv.Rectangle(original,
(car_rect[0], car_rect[1]),
(car_rect[0] + car_rect[2], car_rect[1] + car_rect[3]),
(255, 0, 0),
1,
8,
0)
for point in points:
cv.Circle(original,
point,
1,
(0, 0, 255),
-1,
8,
0)
cv.ShowImage('Analysed', original)
k = cv.WaitKey(33)
can you please tell me how to track 2 or more objects?
ReplyDeleteHi,
ReplyDeleteThe key to finding multiple objects is the cv.FindContours method that robby_rails used. This can be used to find multiple areas of interest. My method was very limited in that it only found a single area of interest, in specific lighting conditions.
Cheers,
Robert
can you please send me the c++ code,beacuse I am working in c++
ReplyDeletemy email is sekhar4233@gmail.com
Hi Sekhar, thank you for your comment but unfortunately I don't have a lot of c++ experience so I wouldn't the best person to help here. I can only suggest having a look over Stack Overflow for helpful examples. Cheers, Robert
ReplyDeleteIt give me this error
ReplyDeleteAttributeError: 'module' object has no attribute 'CaptureFromCAM'
AttributeError: 'module' object has no attribute 'NamedWindow'
AttributeError: 'module' object has no attribute 'SetCaptureProperty'
How can i solve them?