YOLO给每个检测框加序号
时间: 2023-07-06 11:11:33 浏览: 104
目标检测+旋转目标框+YOLO+小目标检测
5星 · 资源好评率100%
可以通过在检测框中加入序号文本来给每个检测框加上序号。具体方法是使用OpenCV库中的putText函数,在每个检测框的中心位置绘制序号文本。在Python中,可以使用以下代码实现:
```python
import cv2
# 读取图片和检测框信息
img = cv2.imread("image.jpg")
detections = [(100, 200, 50, 50), (300, 150, 80, 80), (500, 300, 30, 30)]
# 给每个检测框加上序号
for i, det in enumerate(detections):
x, y, w, h = det
# 计算文本位置
text_x = x + w // 2
text_y = y + h // 2
# 绘制文本
cv2.putText(img, str(i + 1), (text_x, text_y), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
# 显示结果
cv2.imshow("Result", img)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
其中,detections是一个列表,每个元素表示一个检测框的位置和大小,格式为(x, y, w, h)。putText函数的第一个参数是要绘制文本的图像,第二个参数是要绘制的文本内容,第三个参数是文本位置,第四个参数是字体,第五个参数是字体大小,第六个参数是文本颜色,第七个参数是文本粗细。运行以上代码,将在图片中给每个检测框加上序号。
阅读全文