testEstsMask.py 文件源码

def createMask(originalMap , alpha):

    height,width = originalMap.shape;
    threshold = alpha * np.amax(originalMap)

    # im2bw
    bw = np.zeros((height,width))
    idx = originalMap > threshold
    bw[idx] = 1

    location = np.argmax(originalMap)
    [row,col] = np.unravel_index(location,(height,width))
    blobs = bw==1
    ConnectedComponent = measure.label(blobs)
    CompLabel =  ConnectedComponent[row,col]
    PixelList = ConnectedComponent==CompLabel

    mask = np.ones((height,width));
    mask[PixelList] = 0;

    return mask