#仅供自己学习笔记用
主要是前面的img2vector是新的知识点,也许基础知识会,能理解,但是自己写真的能卸出来吗
returnVect要先预设 然后往里面装 装的时候是分行读的,要怎么写? int有注意到吗
import numpy as np
import os
from com.bjsxt.knn.KNNDateOnHand import classify
#就是封装好的包呗
def img2vector(filename):
returnVect=np.zeros((1,1024))#原数据是32*32,而且注意是双重括号
fr=open(filename)
for line in range(32):
lineStr=fr.readline()
for j in range(32):
returnVect[0,32*i+j]=int(lineStr[j])
return returnVect
def IdentifImgClassText():
hwLabel=[]
trainingFileList=os.listdir('TrainData') #https://www.runoob.com/python/os-listdir.html
m=len(trainingFileList)
trainingMat=np.zeros((m,1024))
for i in range(m):
fileNameStr=trainingFileList[i]
fileStr=fileNameStr.split(".")[0]
fileNumStr=int(fileStr.split("_")[0])
hwLabes.append(classNumStr)
trainingMat[i,:]=img2vector('TrainData/%s'%fileNameStr)
#然后也是又写了一遍,没有写循环呀,不是简便的就是好的呀
testFileList=os.listdir('TestData')
errorCount=0.0
mTest=len(testFileList)
for i in range(mTest):
fileNameStr=trainingFileList[i]
fileStr=fileNameStr.split(".")[0]
fileNumStr=int(fileStr.split("_")[0]) #记得要化成整数型
vectorUnderTest=img2vector('TestData/%s'%fileNumStr)
classifierResult=classify(vectorUnderTest,trainingMat,hwLabels,3)
print("识别出来的数字是: %d,真实的数字是:%d"(classifierResult,classNumStr))
if(classifierResult!=classNumStr):
errorCount+=1.0
print("\n识别错误的个数%d"%errorCount)
errorRate=errorCount/float(mTest)
print("\n识别错误率%f"%erroRate)
if_name_=="_main_":
#https://blog.csdn.net/Nancy50/article/details/90205367
IdentifyImgClassTest