MNIST数据集卷积神经网络实现手写数字识别应用(GUI)
代码中GUI实现的并不美观,只是实现出来GUI需求,大家有需要的可以调整一下布局让GUI更加美观。
谢谢B站的朋友们指正代码错误之处!
代码有两种暂时有两种下载方式:
- 直接clone仓库代码
- 从release中下载我打包好的代码
- tensorflow版本为2.0.0及以上应该都可以运行
- 我的python版本为3.7(兼容tensorflow2.0及以上版本的Python版本应该都可以)
- Operation System:Windows10
- IDE:Pycharm
- 演示用的图片我已经放到numbers_images文件夹中了,百度网盘不方便的直接clone整个仓库即可
- 图片是MOOC上《人工智能实践:Tensorflow笔记》北大曹健老师https://www.icourse163.org/course/PKU-1002536002?tid=1452937471 课程中演示的那10张图片。
- CNN-Model.py为卷积神经网络的训练文件。
- gui.py为图像化界面的启动文件
- checkpoin和weights.txt均为卷积神经网络的训练参数。
- recongnition.py和icon.ico不要修改。
- 训练好的checkpoint和weights.txt文件已经在仓库里面了,如果想自己训练可以直接run CNN-Model.py即可,run完也会生成checkpoint和weights.txt文件。
- 直接执行recognition.py文件,出现gui窗口,选择图片识别即可。