yolov8跟踪人代码
时间: 2024-01-04 18:20:36 浏览: 160
以下是使用YOLOv8进行人体目标跟踪的代码示例:
```python
import cv2
from yolov5 import YOLOv5
# 加载YOLOv5模型
model = YOLOv5()
# 加载视频
video_path = "path/to/video.mp4"
cap = cv2.VideoCapture(video_path)
# 循环读取视频帧
while cap.isOpened():
ret, frame = cap.read()
if not ret:
break
# 使用YOLOv5进行目标检测和跟踪
results = model.detect(frame)
# 遍历检测结果
for result in results:
label = result["label"]
confidence = result["confidence"]
bbox = result["bbox"]
# 绘制边界框和标签
cv2.rectangle(frame, (bbox[0], bbox[1]), (bbox[2], bbox[3]), (0, 255, 0), 2)
cv2.putText(frame, f"{label}: {confidence:.2f}", (bbox[0], bbox[1] - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (0, 255, 0), 2)
# 显示结果
cv2.imshow("Tracking", frame)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
# 释放资源
cap.release()
cv2.destroyAllWindows()
```
请注意,上述代码中的`yolov5`是一个自定义的模块,用于加载和使用YOLOv5模型。你需要根据你的实际情况进行相应的模型加载和目标检测操作。
阅读全文