首先计算第一个卷积层的参数量:

每个卷积核的参数量为3x3x3=27(RGB图像有三个通道) 共有64个卷积核,因此第一个卷积层的参数量为27x64=1728

接下来计算第二个卷积层的参数量:

每个卷积核的参数量为5x5x64(上一层有64个通道) 共有128个卷积核,因此第二个卷积层的参数量为5x5x64x128=51200

全连接层的参数量为:

第一个全连接层有128个神经元,每个神经元需要连接上一层的所有128x4x4个神经元(经过两次2x2的平均池化层,特征图大小变为了128x4x4),因此第一个全连接层的参数量为128x128x4x4=262144

输出层的参数量为:

输出层有10个神经元,每个神经元需要连接上一层的所有128个神经元,因此输出层的参数量为10x128=1280

总的参数量为:

1728+51200+262144+1280=317352

因此该网络的参数量为317352。

假设输入是一张128x128的RGB彩色图像如果网络第一个卷积层的卷积核为3x3的大小步长为1共有64个卷积核后接一个步长为2的2x2平均池化层随后再接两层核大小为5×5卷积核个数为128的卷积层以及一层步进为2的2x2的平均池化层最后接含有128个神经元的全连接层和一个含有10个神经元的输出层试计算该网络的参数量。5x5的卷积核步长为1假设卷积过程默认加入Padding卷积之后的特征图尺寸大小不

原文地址: https://www.cveoy.top/t/topic/bb5C 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录