图像分割python代码

以下是使用Python实现图像分割的示例代码：

import cv2
import numpy as np

# 读入图像
img = cv2.imread('input_image.png')

# 转换为灰度图像
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# 应用阈值操作
ret, thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV+cv2.THRESH_OTSU)

# 执行形态学操作
kernel = np.ones((3,3), np.uint8)
opening = cv2.morphologyEx(thresh, cv2.MORPH_OPEN, kernel, iterations=2)

# 执行背景区域标记操作
sure_bg = cv2.dilate(opening,kernel,iterations=3)

# 执行前景区域标记操作
dist_transform = cv2.distanceTransform(opening,cv2.DIST_L2,5)
ret, sure_fg = cv2.threshold(dist_transform,0.7*dist_transform.max(),255,0)

# 执行未知区域标记操作
sure_fg = np.uint8(sure_fg)
unknown = cv2.subtract(sure_bg,sure_fg)

# 执行标记分水岭算法
ret, markers = cv2.connectedComponents(sure_fg)
markers = markers + 1
markers[unknown==255] = 0
markers = cv2.watershed(img,markers)
img[markers == -1] = [255,0,0]

# 显示结果
cv2.imshow('Segmented Image', img)
cv2.waitKey(0)
cv2.destroyAllWindows()

这段代码使用OpenCV库实现了基于阈值和形态学操作的图像分割，包括背景区域、前景区域和未知区域的标记，以及标记分水岭算法的应用。