在上一篇文章,我们介绍了CNN的一些基本概念和lenet神经网络的架构,今天这一篇文章我们就模仿lenet网络去微调,使其符合mnist数据集的要求,并且达到我们练手的目的....因为mnist的数据集的图片大小为28*28*1的灰度图片,而lenet神经网络是32*32*1的输入,因此我们将其微调,结构如下:
?...在conv2d函数中,我们定义了卷积层,我们在TensorFlow中直接使用tf.nn.con2d()这个函数就可以,他的结构如下:
?...在max_pool_2*2()这个函数中,我们定义了池化层,同样我们也使用了TensorFlow中的tf.nn.nax_pool这个函数,这个函数的结构如下:
?...2:从 list 中依次取出矩阵的长宽及深度,并求三者的乘积,得到矩阵被拉长后的 长度。
3:将 pool2 转换为一个 batch 的向量再传入后续的全连接。