分水岭算法分割
处理图片:[my_coin.jpg]
import cv2
import matplotlib.pyplot as plt
import numpy as np
def main():
# 1.导入图片
img_src = cv2.imread("my_coin.jpg")
# 2.转换为灰度图片
img_gray = cv2.cvtColor(img_src, cv2.COLOR_BGR2GRAY)
img_rgb = cv2.cvtColor(img_src, cv2.COLOR_BGR2RGB)
img_show = img_rgb.copy()
# 3.二值化处理
ret, img_bin = cv2.threshold(img_gray, 127, 255, cv2.THRESH_BINARY_INV)
# 4.执行开运算
k = np.ones((3, 3), np.uint8)
img_opening = cv2.morphologyEx(img_bin, cv2.MORPH_OPEN, kernel=k, iterations=2)
img_sure_bg = cv2.dilate(img_opening, k, iterations=3)
# 5.距离计算与前景剥离
dist_transform = cv2.distanceTransform(img_opening, cv2.DIST_L2, 5)
ret, img_fore = cv2.threshold(dist_transform, 0.7 * dist_transform.max(), 255, cv2.THRESH_BINARY)
img_fore = np.uint8(img_fore)
# 6.标注前景图片
ret, img_markers1 = cv2.connectedComponents(img_fore)
img_fore_adv = img_fore.copy()
unknown = cv2.subtract(img_sure_bg, img_fore_adv)
img_markers2 = img_markers1 + 1
img_markers2[unknown == 255] = 0
# 7.分水岭算法分割
img_markers = cv2.watershed(img_rgb, img_markers2)
img_rgb[img_markers == -1] = [0, 255, 0]
# 8.显示结果
plt.figure("显示结果", figsize=(12, 6))
plt.subplot(121)
plt.imshow(img_show)
plt.axis("off")
plt.subplot(122)
plt.imshow(img_rgb)
plt.axis("off")
plt.show()
if __name__ == '__main__':
main()
处理结果图片