代码语言：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)

    # 获取检测结果
    detections = results.xyxy[0].cpu().numpy()

    # 更新SORT跟踪器
    tracked_objects = mot_tracker.update(detections)

    for track in tracked_objects:
        x1, y1, x2, y2, track_id = track.astype(int)
        
     .........完整代码请登录后点击上方下载按钮下载查看

网友评论0

取消

评论提交中...

提交评论