yolov8如何保留预测结果图片
时间: 2023-10-06 11:09:28 浏览: 874
yolov8预测路标的代码、权重、示例图片和对应的推理结果
在YOLOv8中,可以使用以下代码保存预测结果图片:
```python
import cv2
import numpy as np
# Load image
img = cv2.imread("image.jpg")
# Make prediction
predictions = model.predict(img)
# Draw bounding boxes on image
for prediction in predictions:
x, y, w, h = prediction["bbox"]
label = prediction["label"]
confidence = prediction["confidence"]
cv2.rectangle(img, (x, y), (x+w, y+h), (0, 255, 0), 2)
cv2.putText(img, f"{label}: {confidence:.2f}", (x, y-10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)
# Save image
cv2.imwrite("result.jpg", img)
```
这个代码段首先加载输入图像,然后使用模型进行预测。之后,它使用预测结果在图像上绘制边界框和标签,并将结果保存到磁盘上的一个新图像文件中。
阅读全文