OpenCV 目标追踪

    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
    }

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

    vs = cv2.VideoCapture("2.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,h,w) = [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'):
        box = cv2.selectROI("Frame",frame,fromCenter=False,
                           showCrosshair=True)
        tracker = OPENCV_OBJECT_TRACKERS["medianflow"]()
        trackers.add(tracker,frame,box)
        
    elif key ==ord("d"):
        break

    vs.release()
    cv2.destroyAllWindows()