BoundaryExtraction.py 文件源码

def FindUpperBoundary(path,mode,origin_img=False):
    # flag,up_bd,preprocess_img = UpperBoundary(path,mode)
    flag,up_bd,preprocess_img = UpperBoundaryUpdate(path,mode,origin_img=origin_img)
    crop_margin = 50
    up_bd_flip = []
    if flag==False:
        # flag_flip,up_bd_flip,preprocess_img_flip = UpperBoundary(path,mode,flip=True)
        flag_flip,up_bd_flip,preprocess_img_flip = UpperBoundaryUpdate(path,mode,flip=True,origin_img=origin_img)
        if up_bd_flip==False:
            return

    row,col = preprocess_img.shape
    bd_img = np.zeros((row,col),dtype=np.uint8)
    for pos in up_bd_flip:
        pos[1] = col-1-pos[1]
    final_up_bd = LineUpPixels(up_bd,up_bd_flip,col)
    # print(final_up_bd)
    for i in range(len(final_up_bd)):
        bd_img[int(final_up_bd[i])][i] = 255


    return final_up_bd