tracker.py 898 Bytes
Newer Older
1
import numpy as np
2
import cv2 as cv
3 4 5 6 7
import sys

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

9 10
cv.namedWindow("tracking")
camera = cv.VideoCapture(sys.argv[1])
11 12 13 14
ok, image=camera.read()
if not ok:
    print('Failed to read video')
    exit()
15 16
bbox = cv.selectROI("tracking", image)
tracker = cv.TrackerMIL_create()
17 18 19 20 21
init_once = False

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

    if not init_once:
        ok = tracker.init(image, bbox)
        init_once = True

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

    if ok:
        p1 = (int(newbox[0]), int(newbox[1]))
        p2 = (int(newbox[0] + newbox[2]), int(newbox[1] + newbox[3]))
35
        cv.rectangle(image, p1, p2, (200,0,0))
36

37 38
    cv.imshow("tracking", image)
    k = cv.waitKey(1) & 0xff
39
    if k == 27 : break # esc pressed