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

使用Python绘制图像数组上的边界框与颜色标识

发布时间:2023-12-15 19:11:40

Python提供了多种绘制图像的库,比如OpenCV和Pillow。在本例中,我们将使用OpenCV来绘制图像数组上的边界框和颜色标识带。

首先,我们需要安装OpenCV库。可以使用以下命令在终端中安装OpenCV:

pip install opencv-python

接下来,我们将展示一个使用OpenCV绘制图像数组上的边界框和颜色标识带的例子。

import cv2
import numpy as np

# 创建一个空白图像数组
image = np.zeros((300, 300, 3), dtype=np.uint8)

# 绘制一个矩形边界框
cv2.rectangle(image, (50, 50), (200, 200), (0, 255, 0), 2)

# 绘制一个圆形边界框
cv2.circle(image, (250, 150), 50, (0, 0, 255), 3)

# 在图像上添加颜色标识带
cv2.line(image, (0, 250), (300, 250), (255, 0, 0), 5)
cv2.putText(image, "Red", (10, 270), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (255, 0, 0), 2)

cv2.line(image, (0, 280), (300, 280), (0, 255, 0), 5)
cv2.putText(image, "Green", (10, 300), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 255, 0), 2)

cv2.line(image, (0, 320), (300, 320), (0, 0, 255), 5)
cv2.putText(image, "Blue", (10, 340), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 0, 255), 2)

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

在这个例子中,我们首先创建一个空白的图像数组,大小为300x300。然后,我们使用OpenCV的cv2.rectangle()函数绘制一个矩形边界框,并使用RGB颜色(0, 255, 0)来表示绿色。

接下来,我们使用cv2.circle()函数绘制一个圆形边界框,并使用RGB颜色(0, 0, 255)来表示红色。

最后,我们使用cv2.line()函数绘制了三条水平线,并使用RGB颜色(255, 0, 0)、(0, 255, 0)和(0, 0, 255)来表示红色、绿色和蓝色。我们还使用cv2.putText()函数在每个颜色下方添加了相应的文本标识。

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

运行以上代码,将会得到一个绘制了边界框和颜色标识带的图像。您可以根据需要修改颜色、坐标和文本内容来适应不同的场景。