testEstsMask.py 文件源码

python
阅读 16 收藏 0 点赞 0 评论 0

项目:Human-Pose-Estimation-Using-FCN 作者: jessiechouuu 项目源码 文件源码
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
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号