案例来源于© 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()