深度學習模型--卷積參數計算

1.深度學習參數計算代碼

# --coding:utf-8--
from keras.layers import Conv2D, Input, Activation
from keras import Model

def build_model(input_shape, filter_num=16):
    x = Input(input_shape)
    c_1 = Conv2D(filter_num, (3, 3), use_bias=True)(x)
    out = Activation('relu')(c_1)
    model = Model(input=x, output=out)
    return model
    
model = build_model((28, 28, 1), filter_num=16)
# Param = filter_num*kernel_size*kernel_size + filter_num(use_bias=True)
model.summary()

2.計算公式

Param = filter_num * kernel_size * kernel_size + filter_num(use_bias=True)

3.計算截圖

use_bias=False

use_bias=True