multitracker.py 1.15 KB
Newer Older
1
import numpy as np
2
import cv2 as cv
3 4 5 6 7 8 9
import sys

if len(sys.argv) != 2:
    print('Input video name is missing')
    exit()

print('Select 3 tracking targets')
10

11 12 13
cv.namedWindow("tracking")
camera = cv.VideoCapture(sys.argv[1])
tracker = cv.MultiTracker_create()
14 15
init_once = False

16 17 18 19 20
ok, image=camera.read()
if not ok:
    print('Failed to read video')
    exit()

21 22 23
bbox1 = cv.selectROI('tracking', image)
bbox2 = cv.selectROI('tracking', image)
bbox3 = cv.selectROI('tracking', image)
24

25 26 27
while camera.isOpened():
    ok, image=camera.read()
    if not ok:
28
        print 'no image to read'
29 30 31
        break

    if not init_once:
32 33 34
        ok = tracker.add(cv.TrackerMIL_create(), image, bbox1)
        ok = tracker.add(cv.TrackerMIL_create(), image, bbox2)
        ok = tracker.add(cv.TrackerMIL_create(), image, bbox3)
35 36 37 38 39 40 41 42
        init_once = True

    ok, boxes = tracker.update(image)
    print ok, boxes

    for newbox in boxes:
        p1 = (int(newbox[0]), int(newbox[1]))
        p2 = (int(newbox[0] + newbox[2]), int(newbox[1] + newbox[3]))
43
        cv.rectangle(image, p1, p2, (200,0,0))
44

45 46
    cv.imshow('tracking', image)
    k = cv.waitKey(1)
47
    if k == 27 : break # esc pressed