帮我写一个opencv python目标跟踪例子
时间: 2023-03-23 20:04:34 浏览: 117
基于OPENCV的目标跟踪
4星 · 用户满意度95%
好的,我可以为您提供一个基于 OpenCV 和 Python 的目标跟踪示例。您可以使用 OpenCV 库中的 cv2.Tracker 类来实现目标跟踪。以下是一个简单的示例代码:
```
import cv2
# 选择跟踪器类型
tracker_type = "CSRT"
if tracker_type == "BOOSTING":
tracker = cv2.TrackerBoosting_create()
elif tracker_type == "MIL":
tracker = cv2.TrackerMIL_create()
elif tracker_type == "KCF":
tracker = cv2.TrackerKCF_create()
elif tracker_type == "TLD":
tracker = cv2.TrackerTLD_create()
elif tracker_type == "MEDIANFLOW":
tracker = cv2.TrackerMedianFlow_create()
elif tracker_type == "GOTURN":
tracker = cv2.TrackerGOTURN_create()
elif tracker_type == "MOSSE":
tracker = cv2.TrackerMOSSE_create()
else:
tracker = cv2.TrackerCSRT_create()
# 读取视频文件
video = cv2.VideoCapture("test.mp4")
# 选择要跟踪的初始目标
ret, frame = video.read()
bbox = cv2.selectROI(frame, False)
tracker.init(frame, bbox)
# 循环遍历视频帧并进行目标跟踪
while True:
# 读取下一帧
ret, frame = video.read()
if not ret:
break
# 更新跟踪器并获取新的边界框
success, bbox = tracker.update(frame)
# 如果跟踪成功,则在帧上绘制边界框
if success:
(x, y, w, h) = [int(v) for v in bbox]
cv2.rectangle(frame, (x, y), (x + w, y + h), (, 255, ), 2)
# 显示当前帧
cv2.imshow("Frame", frame)
key = cv2.waitKey(1) & xFF
# 如果用户按下 'q' 键,则退出循环
if key == ord("q"):
break
# 清理资源并关闭窗口
video.release()
cv2.destroyAllWindows()
```
希望这个示例能够帮助您实现目标跟踪。
阅读全文