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

Question

我使用skimage.morphology.watershed来获得序列图像的初始分割。但是，我想为分段区域设置一个最大体积

例如，我有一个矩阵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. 

结果： segmentation with watershed

实际上，我是在3D工作的，但2D的原理也应该适用。谢谢你的帮助