python使用opencv的内置跟踪器跟踪摄像头中框选指定的目标代码

代码语言:python

所属分类:其他

代码描述:python使用opencv的内置跟踪器跟踪摄像头中框选指定的目标代码,如果报错,执行下面的命令 pip install --upgrade opencv-contrib-python

代码标签: python 使用 opencv 内置 跟踪器 跟踪 摄像头 框选 指定 目标 代码

下面为部分代码预览,完整代码请点击下载或在bfwstudio webide中打开

#!/usr/local/python3/bin/python3
# -*- coding: utf-8 -*
import cv2

def initialize_tracker(frame):
    """初始化跟踪器并选择ROI"""
    tracker = cv2.TrackerKCF_create()
    roi = cv2.selectROI('Select ROI', frame, False)
    tracker.init(frame, roi)
    return tracker, roi

def main():
    # 初始化视频捕获
    cap = cv2.VideoCapture(0)  # 使用默认摄像头,如果有多个摄像头,可以尝试更改这个数字

    if not cap.isOpened():
        print("无法打开摄像头")
        exit()

    # 读取一帧图像
    ret, frame = cap.read()
    if not ret:
        print("无法获取图像")
        cap.release()
        exit()

    # 初始化跟踪器并选择ROI
    tracker, roi = initialize_tracker(frame)
    tracking = True

    # 主循环
    while True:
        ret, frame = cap.read()
        if not ret:
            print("无法获取图像")
            break

        if tracking:
            # 更新跟踪器
            success, box = tracker.update(frame)
            if success:
                # 绘制边界框
                x, y, w.........完整代码请登录后点击上方下载按钮下载查看

网友评论0