Как рассчитать количество параметров в CNN при работе с картинками

Question

Я получаю вопрос следующим образом:

Предположим, у вас есть цветной ввод изображения 10x10x3 и вы хотите сложить два сверточных слоя с размером ядра 3x3 с 10 и 20 фильтрами соответственно.Сколько параметров вы должны обучить для этих двух слоев?

И я знаю, как решить ситуацию с одним сверточным слоем, что число параметров должно быть (filter.shape[0]*filter.shape[1]*...*filter.shape[n] + bias) * number of filters.
Но яНе знаете, как рассчитать количество параметров в многослойных ситуациях.
Может ли кто-нибудь мне помочь?
Заранее спасибо.

Answer 1

Просто добавьте все параметры из каждого слоя. У вас есть формула для одного слоя:

(filter.shape [0] filter.shape [1] ... * filter.shape [n] + смещение) * количество фильтров

Так что просто рассчитайте это для каждого слоя и сложите.

В вашем примере это даст (исключая смещение):

Layer 1 #params = 3x3x3x10 параметров

Layer 2 #params = 3x3x10x20 параметров

Всего = Слой 1 # параметры + Слой 2 # параметры