PyTorch實現(xiàn)手寫數(shù)字的識別入門小白教程

更新時間：2022年06月01日 10:02:08 作者：B.Bz

這篇文章主要介紹了python實現(xiàn)手寫數(shù)字識別，非常適合小白入門學(xué)習(xí)，本文通過實例圖文相結(jié)合給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下

手寫數(shù)字識別（小白入門）

今早剛剛上了節(jié)實驗課，關(guān)于邏輯回歸，所以手有點刺撓就想發(fā)個博客，作為剛剛?cè)腴T的小白，看到代碼運行成功就有點小激動，這個實驗沒啥含金量，所以路過的大牛不要停留，我怕你們吐槽哈哈。

實驗結(jié)果：

在這里插入圖片描述

1.數(shù)據(jù)預(yù)處理

其實呢，原理很簡單，就是使用多變量邏輯回歸，將訓(xùn)練28*28圖片的灰度值轉(zhuǎn)換成一維矩陣，這就變成了求784個特征向量1個標(biāo)簽的邏輯回歸問題。代碼如下：

#數(shù)據(jù)預(yù)處理
trainData = np.loadtxt(open('digits_training.csv', 'r'), delimiter=",",skiprows=1)#裝載數(shù)據(jù)
MTrain, NTrain = np.shape(trainData)  #行列數(shù)
print("訓(xùn)練集：",MTrain,NTrain)
xTrain = trainData[:,1:NTrain]
xTrain_col_avg = np.mean(xTrain, axis=0) #對各列求均值
xTrain =(xTrain- xTrain_col_avg)/255  #歸一化
yTrain = trainData[:,0]

2.訓(xùn)練模型

對于數(shù)學(xué)差的一批的我來說，學(xué)習(xí)算法真的是太太太扎心了，好在具體算法封裝在了sklearn庫中。簡單兩行代碼即可完成。具體參數(shù)的含義隨隨便便一搜到處都是，我就不班門弄斧了，每次看見算法除了頭暈啥感覺沒有。

model = LogisticRegression(solver='lbfgs', multi_class='multinomial', max_iter=500)
model.fit(xTrain, yTrain)

3.測試模型，保存

接下來測試一下模型，準(zhǔn)確率能達到百分之90，也不算太高，訓(xùn)練數(shù)據(jù)集本來也不是很多。
為了方便，所以把模型保存下來，不至于運行一次就得訓(xùn)練一次。

#測試模型
testData = np.loadtxt(open('digits_testing.csv', 'r'), delimiter=",",skiprows=1)
MTest,NTest = np.shape(testData)
print("測試集：",MTest,NTest)
xTest = testData[:,1:NTest]
xTest = (xTest-xTrain_col_avg) /255   # 使用訓(xùn)練數(shù)據(jù)的列均值進行處理
yTest = testData[:,0]
yPredict = model.predict(xTest)
errors = np.count_nonzero(yTest - yPredict) #返回非零項個數(shù)
print("預(yù)測完畢。錯誤：", errors, "條")
print("測試數(shù)據(jù)正確率:", (MTest - errors) / MTest)

'''================================='''
#保存模型

# 創(chuàng)建文件目錄
dirs = 'testModel'
if not os.path.exists(dirs):
    os.makedirs(dirs)
joblib.dump(model, dirs+'/model.pkl')
print("模型已保存")

https://download.csdn.net/download/qq_45874897/12427896 需要的可以自行下載

4.調(diào)用模型

既然模型訓(xùn)練好了，就來放幾張圖片調(diào)用模型試一下看看怎么樣
導(dǎo)入要測試的圖片，然后更改大小為28*28，將圖片二值化減小誤差。
為了讓結(jié)果看起來有逼格，所以最后把圖片和識別數(shù)字同實顯示出來。

import  cv2
import numpy as np
from sklearn.externals import joblib

map=cv2.imread(r"C:\Users\lenovo\Desktop\[DX6@[C$%@2RS0R2KPE[W@V.png")
GrayImage = cv2.cvtColor(map, cv2.COLOR_BGR2GRAY)
ret,thresh2=cv2.threshold(GrayImage,127,255,cv2.THRESH_BINARY_INV)
Image=cv2.resize(thresh2,(28,28))
img_array = np.asarray(Image)
z=img_array.reshape(1,-1)

'''================================================'''

model = joblib.load('testModel'+'/model.pkl')
yPredict = model.predict(z)
print(yPredict)
y=str(yPredict)
cv2.putText(map,y, (10,20), cv2.FONT_HERSHEY_SIMPLEX,0.7,(0,0,255), 2, cv2.LINE_AA)
cv2.imshow("map",map)
cv2.waitKey(0)

5.完整代碼

test1.py

import numpy as np
from sklearn.linear_model import LogisticRegression
import os
from sklearn.externals import joblib

#數(shù)據(jù)預(yù)處理
trainData = np.loadtxt(open('digits_training.csv', 'r'), delimiter=",",skiprows=1)#裝載數(shù)據(jù)
MTrain, NTrain = np.shape(trainData)  #行列數(shù)
print("訓(xùn)練集：",MTrain,NTrain)
xTrain = trainData[:,1:NTrain]
xTrain_col_avg = np.mean(xTrain, axis=0) #對各列求均值
xTrain =(xTrain- xTrain_col_avg)/255  #歸一化
yTrain = trainData[:,0]

'''================================='''
#訓(xùn)練模型
model = LogisticRegression(solver='lbfgs', multi_class='multinomial', max_iter=500)
model.fit(xTrain, yTrain)
print("訓(xùn)練完畢")

'''================================='''
#測試模型
testData = np.loadtxt(open('digits_testing.csv', 'r'), delimiter=",",skiprows=1)
MTest,NTest = np.shape(testData)
print("測試集：",MTest,NTest)
xTest = testData[:,1:NTest]
xTest = (xTest-xTrain_col_avg) /255   # 使用訓(xùn)練數(shù)據(jù)的列均值進行處理
yTest = testData[:,0]
yPredict = model.predict(xTest)
errors = np.count_nonzero(yTest - yPredict) #返回非零項個數(shù)
print("預(yù)測完畢。錯誤：", errors, "條")
print("測試數(shù)據(jù)正確率:", (MTest - errors) / MTest)

'''================================='''
#保存模型

# 創(chuàng)建文件目錄
dirs = 'testModel'
if not os.path.exists(dirs):
    os.makedirs(dirs)
joblib.dump(model, dirs+'/model.pkl')
print("模型已保存")

運行結(jié)果

在這里插入圖片描述

test2.py

import  cv2
import numpy as np
from sklearn.externals import joblib

map=cv2.imread(r"C:\Users\lenovo\Desktop\[DX6@[C$%@2RS0R2KPE[W@V.png")
GrayImage = cv2.cvtColor(map, cv2.COLOR_BGR2GRAY)
ret,thresh2=cv2.threshold(GrayImage,127,255,cv2.THRESH_BINARY_INV)
Image=cv2.resize(thresh2,(28,28))
img_array = np.asarray(Image)
z=img_array.reshape(1,-1)

'''================================================'''

model = joblib.load('testModel'+'/model.pkl')
yPredict = model.predict(z)
print(yPredict)
y=str(yPredict)
cv2.putText(map,y, (10,20), cv2.FONT_HERSHEY_SIMPLEX,0.7,(0,0,255), 2, cv2.LINE_AA)
cv2.imshow("map",map)
cv2.waitKey(0)

提供幾張樣本用來測試：

在這里插入圖片描述