Python OpenCV 连接不封闭的轮廓

开发环境:Python2.7 + OpenCV2

常常用Canny整理的图片边界,存在不封闭的情况,如下图中圈起来的部分:

Python OpenCV 连接不封闭的轮廓

那么这时,如果我们查找边界,并且填充的边界内部的话,则无法完美的实现,所以必须找个方法封闭这个边界。

网上搜了一些文章、博客,没找到更好的方案。在其中一个文章里发现了一个思路,觉得可行,特记录下来(原文链接找不到了)。

思路是找到这个边界的端点,然后用line函数划线连接两个端点即可。

那么什么是端点,如何定义呢?

遍历图片,遇到白色的点时,以他为中心,周围8个点,如果和是255,则此点必是端点。这是原文关于端点的定义。

 

Python OpenCV 连接不封闭的轮廓

(不太会画,颜色反着看就行)四周的点位置任意,只要有一个即可。

开始我也是按照这个规则,进行了边界封闭,但发现还有一种端点情况,文章中并未提及,因此当遇到时,依旧无法完美封闭。

这种情况是:周边会有2个点,且是紧挨着的(在上面第一个图左下角中就有一个这样的点)

Python OpenCV 连接不封闭的轮廓

那么这时,就需要进一步处理找到这种类型的点。

以下是找端点的主要代码,主要实现了上述2中情况下的端点,仅供参考!

for i in range(1, h - 1):          #遍历每一行
        for j in range(1, w - 1):      #遍历每一列
            if dst[i, j] == 255:      #判断该点是否为bai点,0代表黑点
                r = []
                for y in range(i - 1, i + 2):
                    for x in range(j - 1, j + 2):
                        if y == i and x == j:
                            continue
                        if dst[y, x] == 255:
                            r.append((y, x))
                #
                rLen = len(r)
                if rLen > 0:
                    # only one point, it is peak
                    if rLen ==  1:
                        tool.logDebug('r = {}'.format(r))
                        p.append((j, i))
                    # there are two points and the distance is 1, it is also peak
                    elif rLen == 2:
                        dy = r[0][0] - r[1][0]
                        dx = r[0][1] - r[1][1]
                        t = dx * dx + dy * dy
                        if t == 1:
                            tool.logDebug('r = {}'.format(r))
                            p.append((j, i))
    #
    pLen = len(p)
    tool.logDebug('顶点序列:{}'.format(p))