Opencv-python 视频跟踪

参考：https://blog.csdn.net/weixin_45875105/article/details/111311029

"""
    视频跟踪
"""
import sys
import cv2

def main():
    # 1.创建追踪对象
    # tracker = cv2.TrackerBoosting_create()
    # tracker = cv2.TrackerMIL_create()
    tracker = cv2.TrackerKCF_create()
    # tracker = cv2.TrackerTLD_create()
    # tracker = cv2.TrackerCSRT_create()
    # tracker = cv2.TrackerGOTURN_create()
    # tracker = cv2.TrackerMedianFlow_create()

    # 2.读取视频
    video = cv2.VideoCapture("jt1.mp4")

    # 视频读取失败
    if not video.isOpened():
        print("Could not open video")
        sys.exit()

    # 读取第一帧
    ok, frame = video.read()
    if not ok:
        print('Cannot read video file')
        sys.exit()

    # 3.设置跟踪ROI区域
    # 默认跟踪区域
    # bbox = (287, 23, 86, 320)

    # Uncomment the line below to select a different bounding box
    bbox = cv2.selectROI(frame, False)

    # 初始化跟踪对象
    ok = tracker.init(frame, bbox)

    # 4.循环读取视频并跟踪
    while True:
        # 读取视频图片
        ok, frame = video.read()
        if not ok:
            break

        # 开始计时
        timer = cv2.getTickCount()

        # 更新跟踪结果
        ok, bbox = tracker.update(frame)

        # 计算帧率
        fps = cv2.getTickFrequency() / (cv2.getTickCount() - timer)

        # 绘制结果
        if ok:
            # 跟踪失败
            p1 = (int(bbox[0]), int(bbox[1]))
            p2 = (int(bbox[0] + bbox[2]), int(bbox[1] + bbox[3]))
            cv2.rectangle(frame, p1, p2, (255, 0, 0), 2, 1)
        else:
            # 跟踪成功
            cv2.putText(frame, "Tracking failure detected", (100, 80), cv2.FONT_HERSHEY_SIMPLEX, 0.75, (0, 0, 255), 2)

        # 显示结果与图片
        cv2.putText(frame, "FPS : " + str(int(fps)), (100, 50), cv2.FONT_HERSHEY_SIMPLEX, 0.75, (50, 170, 50), 2)
        cv2.imshow("Tracking", frame)

        # 退出
        k = cv2.waitKey(1) & 0xff
        if k == 27:
            break

if __name__ == '__main__':
    main()

选定目标，然后按空格