分水嶺算法

分水嶺流程.png

import cv2 as cv
import numpy


def watershed_demo():
    # remove noise if any 消除噪聲
    print(src.shape)
    # gray, binary image
    blurred = cv.pyrMeanShiftFiltering(src, 10, 100)
    gray = cv.cvtColor(blurred, cv.COLOR_BGR2GRAY)
    ret, binary = cv.threshold(gray, 0, 255, cv.THRESH_BINARY | cv.THRESH_OTSU)
    cv.imshow("binary image", binary)

    # morphology operation 開操作去除噪點(diǎn)
    kernel = cv.getStructuringElement(cv.MORPH_RECT, (3, 3))
    # morphology binary,2次開操作
    mb = cv.morphologyEx(binary, cv.MORPH_OPEN, kernel, iterations=2)
    sure_bg = cv.dilate(mb, kernel, iterations=3)  # 3次膨脹
    cv.imshow("mor-opt", sure_bg)

    # distance transform
    # distance transform
    # DIST_L1:曼哈頓距離，DIST_L2：歐氏距離, masksize:跟卷積一樣
    # 這是我們獲取的字段距離數(shù)值，對(duì)應(yīng)每個(gè)像素都有，所以數(shù)組結(jié)構(gòu)和圖像數(shù)組一致
    dist = cv.distanceTransform(mb, cv.DIST_L2, 3)
    dist_output = cv.normalize(dist, 0, 1.0, cv.NORM_MINMAX)  # 歸一化的距離圖像數(shù)組 0-1之間標(biāo)準(zhǔn)化
    # cv.imshow("distance-t", dist_output*50)    # 這行代碼運(yùn)行不了

    ret, surface = cv.threshold(dist, dist.max()*0.5, 255, cv.THRESH_BINARY)
    # cv.imshow("surface_bin", surface)   # 個(gè)人運(yùn)行不了
    # 計(jì)算marker
    surface_fg = numpy.uint8(surface)  # 計(jì)算前景
    unknown = cv.subtract(sure_bg, surface_fg)  # 計(jì)算未知區(qū)域
    ret, markers = cv.connectedComponents(surface_fg)
    print(ret)

    # watershed transform 分水嶺變換
    markers = markers + 1   # 用label進(jìn)行控制
    markers[unknown == 255] = 0
    markers = cv.watershed(src, markers=markers)  # 分水嶺的地方就編程-1
    src[markers == -1] = [0, 0, 255]
    cv.imshow("result_image", src)


src = cv.imread("./data/coins.png", cv.IMREAD_COLOR)
cv.namedWindow("girl", cv.WINDOW_AUTOSIZE)
cv.imshow("girl", src)
watershed_demo()

cv.waitKey(0)
cv.destroyAllWindows()

運(yùn)行結(jié)果

分水嶺.png