请写一个yolov5识别单张图片并输出精确度的代码
时间: 2023-04-01 19:02:22 浏览: 199
基于YOLOV5的图像训练及识别一体化工具
5星 · 资源好评率100%
以下是使用 YOLOv5 识别单张图片并输出精确度的 Python 代码:
```python
import torch
import cv2
# 加载 YOLOv5 模型
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
# 读取图片
img = cv2.imread('test.jpg')
# 进行目标检测
results = model(img)
# 输出检测结果和精确度
print(results.pandas().xyxy[])
print(results.pandas().xyxy[]['confidence'].mean())
```
其中,`test.jpg` 是待检测的图片路径,`results.pandas().xyxy[]` 是检测结果,`results.pandas().xyxy[]['confidence'].mean()` 是精确度的平均值。
阅读全文