Python 如何实现最大体积的流域分割

我使用了

skimage.morphology.waterrence

对一系列图像进行初始分割。但是，我想为分段区域设置最大体积

例如，我有一个矩阵

M

：

import numpy as np
from skimage.morphology import watershed
from scipy.ndimage.morphology import distance_transform_edt

M = np.zeros((30, 30))
M[5, 5] = 1
M[25, 25] = 2
M_edt = distance_transform_edt(M==0)
M_seg = watershed(M_edt, markers=M)  # |--> maximum pixels in one region should be, for example, 20 pixels. 

结果:

实际上，我在3D中工作，但2D中的原理也应该应用。感谢您的任何帮助提前