Câți greutăți sunt antrenate într-un strat convoluțional cu o kernel de 5x5 și 10 filtre?

Într-un strat convoluțional cu un nucleu de 5x5 și 10 filtre, numărul de greutăți antrenabile se calculează ca: dimensiunea nucleului înmulțită cu numărul de filtre.

Dacă presupunem că canalul de intrare este unul singur (de exemplu, o imagine alb-negru), atunci:

5 × 5 × 10 = 250 greutăți.

Dacă există mai multe canale de intrare (de exemplu, 3 pentru RGB), formula va fi:

5 × 5 × numărul_de_canalelor_de_intrare × 10.

În plus, fiecare filtru are de obicei o abatere (bias), adică încă 10 parametri (unul pentru fiecare filtru).

Exemplu pentru 3 canale de intrare:

kernel_size = 5
input_channels = 3
filters = 10
weights = kernel_size * kernel_size * input_channels * filters  # 5*5*3*10 = 750
biases = filters  # 10
total_params = weights + biases  # 760