欢迎访问宙启技术站
智能推送

在Python中绘制图像数组上的边界框示例

发布时间:2023-12-15 19:04:45

在Python中,我们可以使用cv2.rectangle()函数来绘制图像数组上的边界框。

下面是一个示例代码,演示了如何在图像数组上绘制边界框:

import cv2

# 读取图像
image = cv2.imread('image.jpg')

# 定义边界框的位置和尺寸
x, y, w, h = 100, 100, 200, 200

# 绘制边界框
cv2.rectangle(image, (x, y), (x+w, y+h), (0, 255, 0), 2)

# 显示图像
cv2.imshow('Image', image)
cv2.waitKey(0)
cv2.destroyAllWindows()

在上面的示例中,我们首先使用cv2.imread()函数读取一张图像。然后,我们定义了一个边界框的位置和尺寸,这里的(x, y)表示边界框的左上角坐标,(w, h)表示边界框的宽度和高度。

接下来,我们使用cv2.rectangle()函数绘制边界框。该函数的 个参数是要绘制边界框的图像数组,第二个参数是边界框的左上角坐标,第三个参数是边界框的右下角坐标,第四个参数是边界框的颜色(在BGR格式中,这里是绿色),第五个参数是边界框的线宽。

最后,我们使用cv2.imshow()函数显示绘制好边界框的图像,并使用cv2.waitKey()cv2.destroyAllWindows()函数等待用户按下键盘上的任意键,然后关闭窗口。

这里提供了一个简单的使用例子,但实际应用中,你可能需要在图像上绘制多个边界框,或者在边界框上添加标签等。你可以根据自己的需求进行修改和扩展。