技術でWiki

文書の過去の版を開いています。もしこのまま保存すると、この文書が最新となります。

====== マックスプーリング ======

<code python>
import numpy as np

def max_pooling(array, kernel_size, stride):
    row_size = array.shape[0] - kernel_size + 1
    col_size = array.shape[1] - kernel_size + 1

output_array = np.zeros((int(row_size/stride), int(col_size/stride)))
    
    for row_index in range(0, row_size, stride):
        for col_index in range(0, col_size, stride):
            # 起点からカーネルフィルタに入るarrayの要素を順次比較し、最大値を得る
            max_value = 0
            for r_index in range(0, kernel_size):
                for c_index in range(0, kernel_size):
                    if array[row_index+r_index][col_index+c_index] > max_value:
                        max_value = array[row_index+r_index][col_index+c_index]
            output_array[int(row_index/stride)][int(col_index/stride)] = max_value

return output_array

# 入力データの作成
array = np.zeros((6, 6))
array[0][0] = 10
array[0][1] = 9
array[1][1] = 8
array[2][0] = 6
array[2][2] = 12
array[4][4] = 17

kernel_size = 2
stride = 2

output = max_pooling(array, kernel_size, stride)

print(output)

</code>

編集の概要