Python Функция CNN im2col не имеет смысла - PullRequest
Новая
       20

Python Функция CNN im2col не имеет смысла

1 голос
/ 12 февраля 2020

В обучении CNN Convolution код функции im2col не понят. 

def im2col(input_data, filter_h, filter_w, stride=1, pad=0):
    N, C, H, W = input_data.shape
    out_h = (H + 2*pad - filter_h)//stride + 1
    out_w = (W + 2*pad - filter_w)//stride + 1
    img = np.pad(input_data, [(0,0), (0,0), (pad, pad), (pad, pad)], 'constant')
    col = np.zeros((N, C, filter_h, filter_w, out_h, out_w))

#I DON'T KNOW!
for y in range(filter_h):
        y_max = y + stride*out_h
        for x in range(filter_w):
            x_max = x + stride*out_w
            col[:, :, y, x, :, :] = img[:, :, y:y_max:stride, x:x_max:stride]

    col = col.transpose(0, 4, 5, 1, 2, 3).reshape(N*out_h*out_w, -1)
    return col

Q1. Я не знаю, почему входные данные (img) преобразуются в шесть измерений (столбец). Почему ты так конвертируешь?

Q2. Python синтаксис не знаком. Итак, я не понимаю синтаксис строк 9-13. Можете ли вы объяснить это в C / C ++ / JAVA?

Я много пытался понять. Помоги мне.

Извините за странную грамматику. : (

Хорошего дня и всем спасибо!

...