Python用cv2实现Matlab的baweraopen函数

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

主要用到的函数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)