yolov5调用外接摄像头检测
时间: 2023-06-16 15:04:29 浏览: 192
要调用外接摄像头进行检测,您需要先安装OpenCV库和pytorch库。然后使用以下代码:
``` python
import cv2
import torch
import numpy as np
# Load YOLOv5 model
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
# Open camera
cap = cv2.VideoCapture(0)
while True:
# Read camera frame
ret, frame = cap.read()
# Convert frame from BGR to RGB
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
# Detect objects using YOLOv5
results = model(frame)
# Display results
cv2.imshow('YOLOv5', results.render())
# Wait for key press
key = cv2.waitKey(1)
# Exit loop if 'q' is pressed
if key == ord('q'):
break
# Release camera and close window
cap.release()
cv2.destroyAllWindows()
```
这个脚本使用OpenCV库打开摄像头,然后使用YOLOv5模型检测摄像头中的物体。最后,它在窗口中显示检测结果。要退出循环,请按“q”键。