python使用yolo实现对摄像头多个对象进行区分标注跟踪代码

代码语言:python

所属分类:其他

代码描述:python使用yolo实现对摄像头多个对象进行区分标注跟踪代码

代码标签: python yolo 摄像头 多个 对象 进行 区分 标注 跟踪 代码

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

#!/usr/local/python3/bin/python3
# -*- coding: utf-8 -*
import cv2
import torch
from sort import *  # 确保你已安装了sort包

# 加载YOLO模型
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)

# 初始化SORT跟踪器
mot_tracker = Sort()

# 设置视频捕获
cap = cv2.VideoCapture(1)

# 颜色字典,用于为每个跟踪ID分配唯一颜色
color_dict = {}

while True:
    ret, frame = cap.read()
    if not ret:
        break

    # YOLOv5检测
    results = model(frame)

    # .........完整代码请登录后点击上方下载按钮下载查看

网友评论0