Kui palju kaalu treenitakse 5x5 tuumaga ja 10 filtriga konvolutsioonikihis?

Konvolutsioonikihis, mille kernel on 5x5 ja filtreid 10, arvutatakse õppitavate kaalude arv järgmiselt: kernel'i suurus korrutatakse filtrite arvuga.

Kui eeldada, et sisendkanal on üks (näiteks must-valge pilt), siis:

5 × 5 × 10 = 250 kaalu.

Kui on mitu sisendkanalit (näiteks 3 RGB jaoks), siis valem on:

5 × 5 × sisendkanalite arv × 10.

Lisaks on iga filtril tavaliselt bias, mis tähendab veel 10 parameetrit (üks iga filtri kohta).

Näide kolme sisendkanali jaoks:

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