YOLOv5代码实现例子

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), (0, 0, 255), 2)
    cv2.putText(img, f'{cls} {conf:.2f}', (x1, y1), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2)

# 显示结果
cv2.imshow('image', img)
cv2.waitKey(0)
cv2.destroyAllWindows()

注意：这只是一个简单的示例，在实际应用中可能需要进行更多的处理。