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