python区域增长分割算法实验代码

Python区域增长分割算法是一种基于像素相似性的图像分割方法，它通过对像素进行聚类，将相似的像素点组成一个区域，从而实现图像的分割。以下是一个简单的实验代码示例：

import cv2
import numpy as np

def region_growing(image, seed):
    height, width = image.shape
    visited = np.zeros_like(image)
    region = np.zeros_like(image)

    stack = []
    stack.append(seed)

    while len(stack) > 0:
        x, y = stack.pop()
        if visited[x, y] == 1:
            continue

        visited[x, y] = 1
        region[x, y] = image[x, y]

        for i in range(-1, 2):
            for j in range(-1, 2):
                if 0 <= x + i < height and 0 <= y + j < width and not (i == 0 and j == 0):
                    if abs(int(image[x, y]) - int(image[x + i, y + j])) < 20:
                        stack.append((x + i, y + j))

    return region

if __name__ == "__main__":
    image_path = "example.jpg"
    image = cv2.imread(image_path, 0)
    seed = (100, 100)  # 选择一个种子点

    segmented_region = region_growing(image, seed)

    cv2.imshow("Original", image)
    cv2.imshow("Segmented Region", segmented_region)
    cv2.waitKey(0)
    cv2.destroyAllWindows()

以上代码通过使用OpenCV库读取图像，并选择一个种子点，然后利用区域增长分割算法对图像进行分割。最后使用OpenCV库展示原始图像和分割后的区域。这是一个简单的Python区域增长分割算法实验代码示例。