python + opencv图像处理——分水岭算法

关注

原创最新推荐文章于 2026-05-27 16:55:29 发布 1.5k 阅读 AI 写同款· GEO 优化 ›

【python】OpenCV—WaterShed Algorithm 分水岭的概念和形成可以通过模拟浸入过程来说明：在每一个局部极小值表面，刺穿一个小孔，然后把整个模型慢慢浸入水中，随着浸入的加深，每一个局部极小值的影响域慢慢向外扩展，在两个集水盆汇合处构筑大坝，即形成分水岭。是 OpenCV 中用于图像平滑处理的一个函数，它基于均值漂移（Mean Shift）算法，并通过图像金字塔的方式来实现。分水岭分割算法的基本思想是把图像看作是测地学上的拓扑地貌，图像中每一点像素的灰度值表示该点的海拔高度，每一个局部极小值及其影响区域称为集水盆，而集水盆的边界则形成分水岭。阅读详情

基于距离的分水岭分割流程：

1、输入图像
2、灰度（消除噪声）
3、二值化
4、距离变换
5、寻找种子
6、生成Marker
7、分水岭变换
8、输出图像

from matplotlib import pyplot as plt 
from cv2 import cv2 as cv
import numpy as np 

def watearshel_demo():
    print(img.shape)
    blured = cv.pyrMeanShiftFiltering(img,10,80)
    #操作步骤  gray/binary image  
    gray = cv.cvtColor(blured,cv.COLOR_BGR2GRAY)#观察灰度处理后的图像好不好，不好就进行滤波
    ret,binary = cv.threshold(gray,0,255,cv.THRESH_BINARY|cv.THRESH_OTSU)
    cv.imshow("binary_image",binary)

    # morphology operation
    kernel = cv.getStructuringElement(cv.MORPH_RECT,(1,1))#结构元素
    mb = cv.morphologyEx(binary,cv.MORPH_OPEN,kernel,iterations=2)#进行2次开操作
    sure_bg = cv.dilate(mb,kernel,iterations=3)#进行3次膨胀
    cv.imshow("mor",sure_bg)

    #distace transform  
    #对mb进行距离变换
    dist = cv.distanceTransform(mb,cv.DIST_L2,3)
    dist_output = cv.normalize(dist,0,1.0,cv.NORM_MINMAX)
    cv.imshow("distanc_transform",dist_output*50)

    ret,surface = cv.threshold(dist,dist.max()*0.6,255,cv.THRESH_BINARY)
    cv.imshow("surface",surface)

    surface_fg = np.uint8(surface)#转成int
    unknow = cv.subtract(sure_bg,surface_fg)
    ret,markers = cv.connectedComponents(surface_fg)
    print(ret)

    # watershed transform
    markers = markers + 1
    markers[unknow == 255] = 0
    markers = cv.watershed(img,markers=markers)
    img[markers==-1] = [0,0,255]#红色
    cv.imshow('result',img)
    #cv.imwrite('C:\\pictures\\Test paper\\6.jpg',img)
    
if __name__ == "__main__":
    #filepath = "C:\\pictures\\Test paper\\unqualified\\2019_01_09__16_09_03_452_A.jpg"
    filepath = "C:\\pictures\\others\\measure2.jpg"
    img = cv.imread(filepath)       # blue green red
    cv.namedWindow("input image",cv.WINDOW_AUTOSIZE)
    cv.imshow("input image",img)

    watearshel_demo()

    cv.waitKey(0)
    cv.destroyAllWindows()

图片如下：
在这里插入图片描述