与matlab里面 imreconstruct 函数相同的python代码

import cv2
import numpy as np

def imreconstruct(marker, mask, SE=np.ones([3, 3])):   # 开重构
    """
    描述：以mask为约束，连续膨胀marker，实现形态学重建，其中mask >= marker
    参数：
        - marker 标记图像，单通道/三通道图像
        - mask   模板图像，与marker同型
        - conn   联通性重建结构元，参照matlab::imreconstruct::conn参数，默认为8联通。默认结构算子是3*3，全1的
    """
    while True:
        marker_pre = marker
        dilation = cv2.dilate(marker, kernel=SE)
        marker = np.min((dilation, mask), axis=0)
        if (marker_pre == marker).all():
            break
    return marker

        没啥要解释的参数看函数的描述就好了，我也测试过，基本可以做到和 matlab 同输入同输出