基于PyTorch的MNIS数字识别——PyQt

引言

最近实在是太无聊了,不知道为啥又突然想搞起PyQT-UI设计来了。这次呢,我来做个MNIS手写数字识别UI设计程序,供给大家参考参考。

正文

设计环境:windows10、Pytorch框架、python语言、PyCharm编辑器和PyQT5。当然了没有PyQT5的话,也没事啊,本次代码就当作新手门的参考例子了。
基于PyTorch的MNIS数字识别——PyQt_第1张图片
可以看到本代码的文件里有多个文件,其中model.py代码里面主要是神经网络模型和训练相关代码组合,test.py为单个图像测试代码,ui.py为UI界面设计。test文件夹为我新增加的,主要是未来ui测试用的。
实现的效果如下:
基于PyTorch的MNIS数字识别——PyQt_第2张图片
如图所示,该界面为初始化界面,点击开始,进入测试界面,如下图所示:
基于PyTorch的MNIS数字识别——PyQt_第3张图片
如图所示:左边的大“8”字为神经网络模型预测输出的结果,右边的“8”图片为此刻下的识别的图片。左下角为表示识别正确个数,而右下角显示的为识别次数。
当预测错误时,则会显示效果如下:
基于PyTorch的MNIS数字识别——PyQt_第4张图片

是不是觉得很有意思啊,参考代码可参考:My GitHub,有用的话记得点赞哦

你可能感兴趣的:(基于PyTorch的MNIS数字识别——PyQt)