基於PyTorch的MNIS數字識別——PyQt

引言

最近實在是太無聊了,不知道爲啥又突然想搞起PyQT-UI設計來了。這次呢,我來做個MNIS手寫數字識別UI設計程序,供給大家參考參考。

正文

設計環境:windows10、Pytorch框架、python語言、PyCharm編輯器和PyQT5。當然了沒有PyQT5的話,也沒事啊,本次代碼就當作新手門的參考例子了。

可以看到本代碼的文件裏有多個文件,其中model.py代碼裏面主要是神經網絡模型和訓練相關代碼組合,test.py爲單個圖像測試代碼,ui.py爲UI界面設計。test文件夾爲我新增加的,主要是未來ui測試用的。
實現的效果如下:
開始界面
如圖所示,該界面爲初始化界面,點擊開始,進入測試界面,如下圖所示:
在這裏插入圖片描述
如圖所示:左邊的大“8”字爲神經網絡模型預測輸出的結果,右邊的“8”圖片爲此刻下的識別的圖片。左下角爲表示識別正確個數,而右下角顯示的爲識別次數。
當預測錯誤時,則會顯示效果如下:
在這裏插入圖片描述

是不是覺得很有意思啊,參考代碼可參考:My GitHub,有用的話記得點贊哦

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章