Расчет количества параметров в сверточном НН - PullRequest
0 голосов
/ 19 мая 2018

Я новичок в исследовании CNN, и я начал с просмотра уроков Andrew'NG.Вот пример, который я не понял:

enter image description here

enter image description here

Как он вычислял # параметры значение?

1 Ответ

0 голосов
/ 20 марта 2019

Как вы можете видеть в Ответе 1 на этот вопрос StackOverflow , формула для вычисления количества параметров сверточной сети имеет вид: channel_in * kernel_width * kernel_height * channel_out + channel_out.

Но эта формула не соответствует вашим данным.И на самом деле чертеж, который вы показываете, не согласуется с таблицей, которую вы даете.

Если я опираюсь на чертеж, то первый CN имеет 3 входных канала, скользящее окно 5 * 5 и 6 выходных данных.каналов, поэтому число параметров должно быть 456.

Вы даете число 208, и это число, полученное для 1 входного канала и 8 выходных каналов (в таблице указано 8, а на рисунке - 6).Таким образом, кажется, что 208 правильно получено из данных таблицы, , если , мы считаем, что существует один входной канал, а не три.

Что касается второго CN, с 6 входными каналами, aВ скользящем окне 5 * 5 и 16 выходных каналов вам нужно 2416 параметров, что выглядит подозрительно близко к 416, числу, указанному в таблице.

Что касается остальных сетей, то это всегда число входных измерений, умноженное начисло выходных измерений плюс один: 5 * 5 * 16 * 120 + 1 = 48 001, 120 * 84 + 1 = 10 081, 84 * 10 + 1 = 841.

...