OpenCV python 分水岭算法分割

分水岭算法分割

处理图片:[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()

处理结果图片