matlab imfill 大小,Python相当于Matlab函数’imfill’的灰度？

这里有两个版本的泛洪填充算法：

第一个,更简单的一个包含两个未定义的变量,但这是一个工作版本：

import numpy as np

import scipy as sp

import scipy.ndimage

def flood_fill(test_array,h_max=255):

input_array = np.copy(test_array)

el = sp.ndimage.generate_binary_structure(2,2).astype(np.int)

inside_mask = sp.ndimage.binary_erosion(~np.isnan(input_array), structure=el)

output_array = np.copy(input_array)

output_array[inside_mask]=h_max

output_old_array = np.copy(input_array)

output_old_array.fill(0)

el = sp.ndimage.generate_binary_structure(2,1).astype(np.int)

while not np.array_equal(output_old_array, output_array):

output_old_array = np.copy(output_array)

output_array = np.maximum(input_array,sp.ndimage.grey_erosion(output_array, size=(3,3), footprint=el))

return output_array