在TensorFlow上使用LeNet-5模型识别MNIST数据
一共三个py程序,mnist_inference.py定义前向传播的过程以及神经网络中的参数,mnist_train.py用于训练和mnist_eval.py用于测试。
注意:使用代码需要修改MNIST数据集存储位置以及卷积网络存储位置。
此代码中的LeNet-5网络为改进版,依然为7层结构,但是使用的第一个卷积层过滤器尺寸为5*5*32,第二个卷积层过滤器尺寸为5*5*64。
注意:训练网络的...
转载
2020-03-24 22:59:47 ·
296 阅读 ·
0 评论