keras搭建卷积神经网络构建模型参数个数计算方式

keras搭建卷积神经网络构建模型参数个数计算方式

keras搭建卷积神经网络构建模型参数个数计算方式_第1张图片

计算过程
(1)用一十二个3x3x3的卷积核对三通道图像150x150x3进行卷积。每一个卷积核有一个偏置。则此层参数个数为(3x3x3+1)x12=336。
(2)经过上面的操作,生成了一十二幅尺寸为150x150x1的图像。则用五个3x3x12的卷积核对单通道图像150x150x1进行卷积。每个卷积核有一个偏置。此层参数的个数为(3x3x12+1)x5=545。(3)卷积后,进行padding补零。后得到五个75x75的单通道的灰度图像。
(4)将图像的每一个像素看成一个神经元,然后平铺开来有神经元个数75x75x5=28125个。(5)上面的28125个神经元与第一个dense层的128个神经元全链接,有未知参数个数(75x75x5+1)x128=3600128个。
(6)上面的128个神经元与最后一个dense层的十个输出神经元全连接,有未知参数个数(128+1)x10=1290个。
(7)因此,总共的未知参数个数为336+545+3600128+1290=3602299个。

计算结果

keras搭建卷积神经网络构建模型参数个数计算方式_第2张图片

你可能感兴趣的:(keras搭建卷积神经网络构建模型参数个数计算方式)