Hány súlyt tanítanak egy 5x5-ös maggal és 10 szűrővel rendelkező konvolúciós rétegben?

Egy 5x5-ös maggal és 10 szűrővel rendelkező konvolúciós rétegben az tanulható súlyok száma a következőképpen számolható: a mag mérete szorozva a szűrők számával.

Ha feltételezzük, hogy a bemeneti csatorna egy (például fekete-fehér kép), akkor:

5 × 5 × 10 = 250 súly.

Ha több bemeneti csatorna van (például 3 RGB-hez), akkor a képlet így néz ki:

5 × 5 × bemeneti_csatornák_száma × 10.

Ezenkívül minden szűrő általában rendelkezik egy eltolással (bias), azaz még 10 paraméter (egy minden szűrőhöz).

Példa 3 bemeneti csatornára:

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