Python opencv Canny 实现掩膜生成

# ===========================canny边缘提取左右形成掩膜===============
def mask(edgs):
    # 边缘化图像
    # edge = cv2.Canny(inputs, 30, 60)
    h, d = edgs.shape
    result_mask = np.zeros((h, d))
    for i in range(h):
        edgeind = np.where(edgs[i] > 0)[0] # 寻找大于0，即白色边界
        if len(edgeind) > 0:
            left = edgeind[0]
            right = edgeind[-1]
            result_mask[i, left:right+1] = 1
    return np.uint8(result_mask*255)