实现平均池化,窗口大小为k*k,stride为1
arr=[[1,2,3,5],[4,5,6,7],[3,2,1,1],[2,1,2,1]]
def averagePooling(arr,k):
N=len(arr)
tot=0
ans=[]
ks=k*k
tot1=sum([arr[i][j] for i in range(k) for j in range(k)])
print(tot1)
#先算每一行的第一个窗口平均池化后的值,从第二行开始使用前一行的计算结果
ans.append([tot1/ks])
for j in range(N-k):
tot1+=sum(arr[j+k][:k])
tot1-=sum(arr[j][:k])
ans.append([tot1/ks])
#计算对应行每一列的窗口平均池化值,每次使用前一列的计算结果。第一列的结果已经通过上面的代码保存了
for i in range(N-k+1):
tot1=ans[i][0]*ks
for j in range(N-k):
tot1+=sum(arr[z][j+k] for z in range(i,i+k))
tot1-=sum(arr[z][j] for z in range(i,i+k))
ans[i].append(tot1/ks)
return ans
print(averagePooling(arr,2))