Сверточная нейронная сеть только для кучи - PullRequest
0 голосов
/ 10 октября 2019

Я пытаюсь реализовать этот проект: https://github.com/Alescontrela/Numpy-CNN

Он хорошо работает с данными в том же формате, что и набор данных MNIST, и с тем же разрешением 28x28. Однако, когда я пытаюсь обучить его изображениям с более высоким разрешением (я изменил Wight изображения в необходимых функциях), это дает мне несколько ошибок.

CNN\utils.py", line 67, in predict
z = w3.dot(fc) + b3 # first dense layer
ValueError: shapes (128,16928) and (800,1) not aligned: 16928 (dim 1) != 800 (dim 0)

Я проследил его обратно к CNN \ network, line204

## Initializing all the parameters
f1, f2, w3, w4 = (num_filt1 ,img_depth,f,f), (num_filt2 ,num_filt1,f,f), (128,800), (10, 128)

Проблема в том, что я не понимаю, почему w3 и w4 имеют такие значения и почему они работают с изображениями 28x28, а не с 100x100.

Если кто-то может помочь мне понять, чтои как заставить программу работать с большими изображениями, я был бы благодарен:)

...