1. 程式人生 > 其它 >Python 計算八鄰域之和及影象連通區域標記

Python 計算八鄰域之和及影象連通區域標記

#Python計算矩陣8鄰域之和
Neighbors = [(1, 1), (1, -1), (1, 0), (-1, 0), (-1, 1), (-1, -1), (0, 1), (0, -1)]
 
def get_neighbors_sum(array_of_arrays, row, col):
    sum_neighbors = 0
    for neighbor in Neighbors:
        dr, dc = neighbor
        try:
            sum_neighbors += array_of_arrays[row+dr][col+dc]
        
except IndexError: pass return sum_neighbors array_of_arrays = #"Matrix" rows = len(array_of_arrays) cols = len(array_of_arrays[0]) for row in range(rows): for col in range(cols): sum_neighbors = get_neighbors_sum(array_of_arrays, row, c

影象連通區域標記

def countRegion(img):
    [high,width] 
= np.shape(img) mask = np.zeros_like(img) mark = 0 union = {} for i in range (high): for j in range(width): if i==0 and j==0: if img[i][j]==255: mark=mark+1 mask[i][j]=mark union[mark]=mark
if i==0 and j!=0: if img[i][j]==255: left = mask[i][j-1] if left!=0: mask[i][j]=left else: mark = mark +1 mask[i][j]=mark union[mark]=mark if j==0 and i!=0: if img[i][j]==255: up = mask[i-1][j] up_right = mask[i-1][j+1] if up==0 and up_right==0: mark = mark+1 mask[i][j]=mark union[mark]=mark if up==0 and up_right!=0: mask[i][j]=up_right if up_right==0 and up!=0: mask[i][j]=up if up!=0 and up_right!=0: if up==up_right: mask[i][j]=up else: mi = min(up,up_right) mask[i][j]=mi if up<up_right: union[up_right]=up else: union[up]=up_right if i!=0 and j!=0: if img[i][j]==255: up = mask[i-1][j] up_left = mask[i-1][j-1] left = mask[i][j-1] up_right = 0 if j+1<width: up_right = mask[i-1][j+1] ma = max(max(max(up,up_left),up_right),left) if ma==0: mark = mark+1 mask[i][j]=mark union[mark]=mark else: if up==up_right and up_right==up_left and up==left: mask[i][j]=up else: mi = min(min(min(up, up_left), up_right), left) if mi!=0: mask[i][j]=mi if up!=mi: union[up]=mi if up_right!=mi: union[up_right]=mi if up_left!=mi: union[up_left]=mi if left!=mi: union[left]=mi else: n_zero = [] if up!=0: n_zero.append(up) if up_left!=0: n_zero.append(up_left) if up_right!=0: n_zero.append(up_right) if left!=0: n_zero.append(left) mi1 = min(n_zero) mask[i][j]=mi1 for it in n_zero: if it!=mi1: union[it]=mi1 for i in range(high): for j in range(width): key = mask[i][j] if key!=0: while union[key]!=key: key = union[key] mask[i][j]=key return mask