37 lines
826 B
Python
37 lines
826 B
Python
|
from skimage.measure import label
|
|||
|
|
import numpy as np
|
|||
|
|
|
|||
|
|
def largestConnectComponent(bw_img, neighbor=2):
|
|||
|
|
'''
|
|||
|
|
compute largest Connect component of a binary image
|
|||
|
|
|
|||
|
|
Parameters:
|
|||
|
|
---
|
|||
|
|
|
|||
|
|
bw_img: ndarray
|
|||
|
|
binary image
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
---
|
|||
|
|
|
|||
|
|
lcc: ndarray
|
|||
|
|
largest connect component.
|
|||
|
|
|
|||
|
|
Example:
|
|||
|
|
---
|
|||
|
|
>>> lcc = largestConnectComponent(bw_img)
|
|||
|
|
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
labeled_img, num = label(bw_img, connectivity=neighbor, background=0, return_num=True)
|
|||
|
|
# plt.figure(), plt.imshow(labeled_img, 'gray')
|
|||
|
|
|
|||
|
|
max_label = 0
|
|||
|
|
max_num = 0
|
|||
|
|
for i in range(1, num+1): # 这里从1开始，防止将背景设置为最大连通域
|
|||
|
|
if np.sum(labeled_img == i) > max_num:
|
|||
|
|
max_num = np.sum(labeled_img == i)
|
|||
|
|
max_label = i
|
|||
|
|
lcc = (labeled_img == max_label)
|
|||
|
|
|
|||
|
|
return lcc
|