文章代码根据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)