LeNet-5的TensorFlow实现源码
利用tensorflow实现手写体数字实现，使用MNIST数据集进行训练，在MNIST测试集上，准确率可到99.1%代码文件说明config.py:神经网络的超参数配置leNet.py:构建LeNet网络Train.py:加载MINIST数据并训练Inference.py:加载模型完成对图片的识别UI.py生成可进行手写文字并预测的界面数据文件说明MNIST_data/:从MNIST上下载的数据集checkpoint:保存的模型文件使用说明在config.py中进行超参数的配置，运行Train.py可以直接进行训练，也可以直接运行UI.py进行验证UI中，用鼠标在黑色区域进行手写，点击检测即可进行识别
文件大小：3.79KB
评论区