Python OpenCV画框|极客笔记

Python OpenCV画框

在计算机视觉领域，边框是一种常见的图像处理技术，用于突出显示对象或区域。OpenCV是一个广泛应用于计算机视觉和图像处理方面的开源库，它提供了丰富的图像处理函数和工具，可以用来实现边框的绘制。本文将介绍如何使用Python中的OpenCV库来绘制边框。

1. 导入OpenCV库

首先，我们需要导入OpenCV库，如果你的Python环境中没有安装OpenCV库，可以使用以下命令进行安装：

pip install opencv-python

导入OpenCV库的代码如下：

import cv2

2. 读取图像

接下来，我们需要读取一张待处理的图像，可以使用OpenCV提供的cv2.imread()函数来读取图像。下面是读取图像的代码示例：

# 读取图像
image_path = "example.jpg"
image = cv2.imread(image_path)

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

在上面的代码中，首先指定了待处理的图像路径，然后使用cv2.imread()函数读取图像并将其存储在image变量中。最后使用cv2.imshow()函数显示读取的图像，并使用cv2.waitKey(0)和cv2.destroyAllWindows()函数暂停程序并关闭窗口。

3. 绘制边框

接下来我们将介绍如何使用OpenCV库来绘制边框。OpenCV提供了cv2.rectangle()函数用于绘制矩形边框，其语法如下：

cv2.rectangle(image, pt1, pt2, color, thickness)

image：待绘制边框的图像。
pt1：矩形的左上角坐标。
pt2：矩形的右下角坐标。
color：边框的颜色，一般以BGR格式指定。
thickness：边框的厚度，如果设置为-1，则表示填充整个矩形。

下面是一个绘制矩形边框的示例代码：

# 绘制矩形边框
pt1 = (100, 100)
pt2 = (300, 300)
color = (0, 255, 0)  # 绿色
thickness = 2
cv2.rectangle(image, pt1, pt2, color, thickness)

# 显示绘制边框后的图像
cv2.imshow("Image with Rectangle", image)
cv2.waitKey(0)
cv2.destroyAllWindows()

在上面的示例代码中，我们首先定义了矩形左上角和右下角的坐标，并指定了边框的颜色和厚度，然后调用cv2.rectangle()函数绘制边框。最后使用cv2.imshow()函数显示绘制边框后的图像。

4. 示例代码

接下来，我们将展示一个完整的示例代码，该代码读取一张图像并绘制多个矩形边框：

import cv2

# 读取图像
image_path = "example.jpg"
image = cv2.imread(image_path)

# 绘制多个矩形边框
rectangles = [
    ((100, 100), (200, 200), (0, 255, 0), 2),
    ((300, 300), (400, 400), (0, 0, 255), 2),
    ((500, 500), (600, 600), (255, 0, 0), 2)
]

for rect in rectangles:
    pt1, pt2, color, thickness = rect
    cv2.rectangle(image, pt1, pt2, color, thickness)

# 显示绘制边框后的图像
cv2.imshow("Image with Rectangles", image)
cv2.waitKey(0)
cv2.destroyAllWindows()

在上面的示例代码中，我们首先定义了多个矩形边框的参数列表rectangles，然后使用循环依次绘制每个矩形边框。最后使用cv2.imshow()函数显示绘制边框后的图像。