漫水填充&ROI-OpenCV入门教程

人工智能

2022-12-04 10:12:49

使用 OpenCV 实现图像填充：漫水填充的指南

理解填充和漫水填充

计算机视觉领域的一句名言是：“没有填充，就没有 OpenCV”。填充是指将图像或图形中的一个封闭区域用某种颜色或图案填满。其中，漫水填充是一种非常常用的填充算法。它从一个种子点开始，通过递归的思想逐步向周围扩散，最终填充整个封闭区域。

ROI（感兴趣区域）

ROI（感兴趣区域）是指图像或图形中一个特定的区域。在 OpenCV 中，我们可以通过掩码或蒙版来指定 ROI。掩码和蒙版都是二进制图像，白色区域表示感兴趣区域，黑色区域表示不感兴趣区域。

使用 OpenCV 实现漫水填充

步骤 1：创建掩码或蒙版

我们可以使用 OpenCV 的 createMask() 函数来创建掩码或蒙版。例如，以下代码创建一个 3x3 的掩码，将中间的像素设置为白色，其余像素设置为黑色：

import cv2

mask = cv2.createMask(3, 3, cv2.CV_8UC1)
mask[1, 1] = 255

步骤 2：创建种子点

种子点是漫水填充算法的起点。我们可以使用 cv2.circle() 函数创建一个圆形种子点。例如，以下代码创建一个位于图像中心、半径为 5 的圆形种子点：

seed_point = (int(img.shape[1] / 2), int(img.shape[0] / 2))
cv2.circle(img, seed_point, 5, (0, 255, 0), -1)

步骤 3：使用漫水填充算法

我们可以使用 cv2.floodFill() 函数来实现漫水填充。cv2.floodFill() 函数的第一个参数是图像，第二个参数是种子点，第三个参数是填充颜色，第四个参数是掩码或蒙版，第五个参数是边界检测标志，第六个参数是迭代次数限制。例如，以下代码使用漫水填充算法将图像中 ROI 区域填充为蓝色：

cv2.floodFill(img, mask, seed_point, (255, 0, 0), (0, 0, 0), (0, 0, 0), 8)

代码示例

以下是一个完整的 OpenCV 漫水填充代码示例：

import cv2

# 加载图像
img = cv2.imread("image.jpg")

# 创建掩码
mask = cv2.createMask(img.shape[0], img.shape[1], cv2.CV_8UC1)
mask[100:200, 100:200] = 255

# 创建种子点
seed_point = (150, 150)

# 使用漫水填充算法
cv2.floodFill(img, mask, seed_point, (255, 0, 0), (0, 0, 0), (0, 0, 0), 8)

# 显示填充后的图像
cv2.imshow("Filled Image", img)
cv2.waitKey(0)
cv2.destroyAllWindows()