创建追踪器方法对象

案例来源于© Fu Xianjun. All Rights Reserved

标题 创建追踪器方法对象

Powershell安装包
pip install opencv-contrib-python==3.4.3.18

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,
    "medianflow": cv2.TrackerMedianFlow_create,
    "mosse": cv2.TrackerMOSSE_create,
}

tracker = cv2.MultiTracker_create()#实例化opencv的tracker

vs = cv2.VideoCapture("2.mp4")
while True:
	#取当前帧
    frame = vs.read()
    
    frame = frame[1]
    #到头就结束
    if frame is None:
        break
    key = cv2.waitKey(100) & 0xFF
    if key == 27:
        break
vs.release()
cv2.destroyAllWindows()

import cv2
import numpy as np

OPENCV_OBJECT_TRACKERS = {
    "csrt": cv2.TrackerCSRT_create,
    "kfc": cv2.TrackerKCF_create,
    "boosting": cv2.TrackerBoosting_create,
    "mil": cv2.TrackerMIL_create,
    "tld": cv2.TrackerTLD_create,
    "medianflow": cv2.TrackerMedianFlow_create,
    "mosse": cv2.TrackerMOSSE_create,
}
trackers = cv2.MultiTracker_create()#实例化opencv的tracker
vs = cv2.VideoCapture("2.mp4")
while True:
	#取当前帧
    frame = vs.read()
    frame = frame[1]
    #到头就结束
    if frame is None:
        break
     #resize每一帧
    (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+w,y+h),(0,255,0),2)
     #显示
    cv2.imshow("frame",frame)
    key = cv2.waitKey(100)& 0xFF
    if key == ord("s"):
    #选择一个区域。按s
        box = cv2.selectROI("frame",frame,fromCenter=False,showCrosshair=True)
        #创建一个新的追踪器
        tracker = OPENCV_OBJECT_TRACKERS["csrt"]()
        trackers.add(tracker,frame,box)
    if key ==27:#按esc退出
        break
vs.release()
cv2.destroyAllWindows()