YOLOv5是一种目标检测算法,其代码实现可以在GitHub上找到。以下是一个使用YOLOv5进行目标检测的示例代码:
import cv2
import torch
from models import YOLOv5
# 加载预训练模型
model = YOLOv5('yolov5s.pt')
# 读取图像
img = cv2.imread('image.jpg')
# 进行目标检测
outputs = model(img)
# 处理输出结果
for output in outputs:
x1, y1, x2, y2, conf, cls = output
cv2.rectangle(img, (x1, y1), (x2, y2), (