yolov5detect的输出结果进行目标框提取
时间: 2023-09-09 15:14:12 浏览: 76
yolov5_目标检测
可以使用以下代码来提取目标框:
```
import cv2
import numpy as np
# 加载图片
img = cv2.imread('test.jpg')
# 加载yolov5detect输出结果
results = np.loadtxt('yolov5detect_results.txt')
# 遍历结果
for result in results:
# 提取目标框
x, y, w, h, confidence, cls = result
left = int(x - w/2)
top = int(y - h/2)
right = int(x + w/2)
bottom = int(y + h/2)
# 绘制目标框
cv2.rectangle(img, (left, top), (right, bottom), (0, 255, 0), 2)
# 显示图片
cv2.imshow('image', img)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
在上面的代码中,我们首先加载了yolov5detect输出结果,然后遍历结果,提取每个目标框的坐标和置信度等信息,最后使用cv2.rectangle函数绘制目标框。最终,我们可以使用cv2.imshow函数显示图片。
阅读全文