python numpy 手写maxpooling，超简洁写法

最新推荐文章于 2024-04-12 08:25:29 发布

小星星么么哒

最新推荐文章于 2024-04-12 08:25:29 发布

阅读量1.4k

点赞数

文章标签： python numpy 机器学习

本文链接：https://blog.csdn.net/qq_41111734/article/details/125529575

版权

python numpy 手写maxpooling，超简洁写法

简洁易懂

def max_pooling(image, pooling_size, stride):
    c,h,w = image.shape()
    p_h,p_w = pooling_size.shape()

    out_h = (h-p_h)//stride+1
    out_w = (w-p_w)//stride+1
    out = np.zeros((c,out_h, out_w))

    for i in range(c):
        for x in range(out_h):
            for y in range(out_w):
                out[i][x][y] = np.max(image[i][x*stride:(x+1)*stride][y*stride:(y+1)*stride])
    return out