参数个数计算是搭建神经网络的关键,错误的参数个数,将直接导致神经网络运算的崩溃,这里简要记录下,供大家参考:
个人设计神经网络时的计算方法:
已知:
输入图像尺寸:m×n×c(c:通道数)
卷积核尺寸:f×f×c
padding 大小:p
stride 大小:s
由此,卷积后的输出图像尺寸为:(非整数时,向下取整)
TensorFlow中的参数个数计算(两种 padding 方式)
已知:
输入图像尺寸:m×n×c(c:通道数)
卷积核尺寸:f×f×c
padding 大小:SAME 和 VALID
stride 大小:s
padding = SAME 时:(非整数时,向上取整)
padding = VALID 时:(非整数时,向上取整)