Yolov5+deepsort的代码

以下是使用YoloV5和DeepSort进行目标跟踪的代码示例：

import cv2
import torch
from deep_sort import build_tracker
from deep_sort.utils.parser import get_config
from yolov5.utils.datasets import LoadStreams, LoadImages
from yolov5.models.experimental import attempt_load

# 加载YoloV5模型
model = attempt_load(weights='yolov5s.pt', map_location=torch.device('cpu'))

# 加载DeepSort跟踪器配置
cfg = get_config()
cfg.merge_from_file('deep_sort/configs/deep_sort.yaml')
deepsort = build_tracker(cfg, use_cuda=False)

# 加载视频
cap = cv2.VideoCapture('test.mp4')

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

    # 运行YoloV5模型
    results = model(frame)

    # 解析检测结果
    detections = []
    for result in results.pred:
        boxes = result[:, :4].cpu().numpy()
        scores = result[:, 4].cpu().numpy()
        labels = result[:, 5].cpu().numpy()
        detections.extend(zip(boxes, scores, labels))

    # 运行DeepSort跟踪器
    outputs = deepsort.update(detections, frame)

    # 可视化结果
    for output in outputs:
        bbox = output['bbox']
        id = output['id']
        cv2.rectangle(frame, (int(bbox[0]), int(bbox[1])), (int(bbox[2]), int(bbox[3])), (0, 255, 0), 2)
        cv2.putText(frame, str(id), (int(bbox[0]), int(bbox[1]) - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)

    cv2.imshow('frame', frame)
    if cv2.waitKey(1) == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

请注意，您需要安装YoloV5和DeepSort库以使此代码正常运行。