0

LeNet的Tensorflow实现

LetNet是Yann Lecun教授开发的一种使用卷积神经网络的手写识别网络,其训练数据是MNIST。网络的结构如下图所示

letnet

上图已经详细告诉了LetNet的拓扑结构,那么利用Tensorflow就可以像搭建积木一样轻松地构建LeNet,代码如下

关于代码,这里的关键是tf.nn.conv2d函数,在使用的时候注意每个参数的设置,其他的难度都不会太大,最后得到的测试数据的准确率为$95.9%$。完整的代码的地址在我的github上:https://github.com/GoingMyWay/OpenCourse/blob/master/Udacity/DeepLearning/notMNIST/4_convolutions.ipynb。不过需要注意的是,这里数据我是用的是notMNIST

goingmyway

我是一只野生程序猿,我关注机器学习,神经网络,深度学习,增强学习,人工智能,Python,C/C++,Linux

发表评论

电子邮件地址不会被公开。 必填项已用*标注