十六.CSRT 追踪算法的使用_trackercsrt

© Fu Xianjun. All Rights Reserved.

import cv2
import numpy as np

OPENCV_OBJECT_TRACKERS = {
    "csrt":cv2.TrackerCSRT_create,
    "kcf":cv2.TrackerKCF_create,
    "boosting":cv2.TrackerBoosting_create,
    "mil":cv2.TrackerMIL_create,
    "tld":cv2.TrackerTLD_create,
}

trackers = cv2.MultiTracker_create()

vs = cv2.VideoCapture("1.mp4")
while True:
    frame = vs.read()
    frame = frame[1]
    if frame is None:
        break
   
    (h,w) = frame.shape[:2]
    width =  600
    r = width/float(w)
    dim = (width,int(h*r))
    frame = cv2.resize(frame,dim)
    #追踪结果
    (success,boxes) = trackers.update(frame)
    for box in boxes:
        (x,y,w,h) = [int(v) for v in box]
        cv2.rectangle(frame,(x,y),(x