报错No module named cv2处理目标跟踪opencv-contrib-python

最新推荐文章于 2025-04-15 15:01:49 发布

柚子同学99

最新推荐文章于 2025-04-15 15:01:49 发布

阅读量347

点赞数 1

文章标签： opencv python

本文链接：https://blog.youkuaiyun.com/weixin_46302766/article/details/120595674

版权

这段代码展示了如何在Python的Anaconda环境中利用OpenCV库实现多种对象追踪算法，包括CSRT、KCF等。通过读取视频文件，选择ROI（感兴趣区域），应用追踪器并显示追踪结果。用户可以通过按下's'键添加新的追踪目标，按ESC键退出程序。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在anaconda环境下载opencv-contrib-python

修改代码如下

import cv2
# opencv已经实现了的追踪算法
OPENCV_OBJECT_TRACKERS = {
	"csrt": cv2.TrackerCSRT_create,
	"kcf": cv2.TrackerKCF_create,
	"boosting": cv2.legacy.TrackerBoosting_create,
	"mil": cv2.TrackerMIL_create,
	"tld": cv2.legacy.TrackerTLD_create,
	"medianflow": cv2.legacy.TrackerMedianFlow_create,
	"mosse": cv2.legacy.TrackerMOSSE_create
}

import numpy as np

cap = cv2.VideoCapture("D:/vision/111.mp4")
trackers = cv2.legacy.MultiTracker_create()
while True:
    _, frame = cap.read()
    if frame is None:
        break
    frame = cv2.resize(frame, (600, int(frame.shape[0] * 600 / frame.shape[1])), cv2.INTER_AREA)
    (success, boxes) = trackers.update(frame)
    for box in boxes:
        (x1, y1, w, h) = box
        cv2.rectangle(frame, (int(x1), int(y1)), (int(x1 + w), int(y1 + h)), (0, 255, 0))
    cv2.imshow("video", frame)
    key = cv2.waitKey(100)
    if key == ord('s'):
        box = cv2.selectROI("video", frame, True)
        trackers.add(cv2.TrackerKCF_create(), frame, box)
    if key == 27:
        break
cap.release()
cv2.destroyAllWindows()