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 同输入同输出