Python用cv2实现Matlab的baweraopen函数

冷不防 2022-05-30 05:52 198阅读 0赞

### 文章代码根据c++改编（来自[http://www.opencv.org.cn/forum.php?mod=viewthread&tid=2865727][http_www.opencv.org.cn_forum.php_mod_viewthread_tid_2865727]） ###

主要用到的函数[cv2.connectedComponentsWithStats()][cv2.connectedComponentsWithStats]  
【代码】

import cv2
    import numpy as np
    #np.set_printoptions(threshold=np.nan)
    def baweraopen(image,size):
        ''' @image:单通道二值图，数据类型uint8 @size:欲去除区域大小(黑底上的白区域) '''
        output=image.copy()
        nlabels, labels, stats, centroids = cv2.connectedComponentsWithStats(image)
        for i in range(1,nlabels-1):
            regions_size=stats[i,4]
            if regions_size<size:
                x0=stats[i,0]
                y0=stats[i,1]
                x1=stats[i,0]+stats[i,2]
                y1=stats[i,1]+stats[i,3]
                for row in range(y0,y1):
                    for col in range(x0,x1):
                        if labels[row,col]==i:
                            output[row,col]=0
        return output

【测试】

im1=cv2.imread('./map.png',0)
    im2=baweraopen(im1,200)
    im3=baweraopen(im1,2000)
    cv2.imshow('im1',im1)
    cv2.imshow('im2',im2)
    cv2.imshow('im3',im3)
    cv2.waitKey(0)

![这里写图片描述][70]  
![这里写图片描述][70 1]  
![这里写图片描述][70 2]

[http_www.opencv.org.cn_forum.php_mod_viewthread_tid_2865727]: http://www.opencv.org.cn/forum.php?mod=viewthread&tid=2865727
[cv2.connectedComponentsWithStats]: https://www.programcreek.com/python/example/89340/cv2.connectedComponentsWithStats
[70]: /images/20220530/935e1b170fce4f3ab926d85648ab269e.png
[70 1]: /images/20220530/8071206b60604567a27965d9e88f6848.png
[70 2]: /images/20220530/406aaf3cf41b4039b1f20a5958cac905.png