pytorch學習之路一:識別MINIST手寫數字數據集

1.首先,參考pytorch官網安裝pytorch,我選擇的是windows系統+python3.7+pip安裝,所以選擇好後,直接在cmd中運行圖中的命令行,pytorch官網:https://pytorch.org/

2.打開pycharm。新建工程,有的童鞋可能會找不到下載的pytorch包,請參考:https://blog.csdn.net/weixin_39954922/article/details/105606956

3.開始擼代碼:(代碼見文末網址)

3.1首先導入需要的包,

3.2 導入MINIST數據集,如果沒有的話可以通過代碼下載,但是真的灰常慢~,還是科學上網好(手動滑稽),這裏說明一下,pycharm下載MINIST數據集後,會自動放在當前工程文件夾下,鏈接裏已下載好共各位看官食用

3.3開始構件卷積神經網絡。這裏使用class類來建立

3.4選擇優化器和損失函數

3.5開始訓練,並保存模型

3.6訓練一次得到模型之後,沒有必要重複訓練,只需要調用模型即可,所以可以將3.5節的代碼註釋掉,但是這裏需要注意的是,調用模型時,仍然需要用到前面定義的cnn,因此,若重新打開一個.py文件調用模型,需要重新聲明一下 CNN class並定義cnn=CNN(),,爲了看清圖片,本代碼中用了opencv-python的imshow()函數來顯示要識別的圖片

3.6預測結果:

 

代碼GITHUB傳送門:https://github.com/1240117300/MINIST

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