手写maxpooling

CVplayer111

于 2022-11-25 12:01:07 发布

阅读量275

点赞数

文章标签： python numpy

本文链接：https://blog.csdn.net/slamer111/article/details/128035711

版权

这段代码定义了一个名为`maxpooling`的函数，它接受一个3D数组（假设为图像通道）作为输入，并使用2x2的最大池化窗口进行下采样。通过遍历输入数组的每个通道和每个二维平面，找到每个2x2区域的最大值并存储到结果矩阵中。最后返回的是一个经过最大池化处理后的3D数组。

摘要由CSDN通过智能技术生成

import numpy as np
def maxpooling(a):
    result=[]
    for k in range(3):
        result_channel=[]
        for i in range(0,10,2):
            result_row=[]
            for j in range(0,10,2):
                b=a[k,i:i+2,j:j+2]
                ans=np.max(b)
                result_row.append(ans)
            result_channel.append(result_row)
        result.append(result_channel)
    return np.array(result)
a=np.array([i for i in range(300)])
a=a.reshape((3,10,10))
print(maxpooling(a))