自己做的手写数字样本及knn分类代码

import os,re,cv2 import numpy as np def createTrainLabels(train_img_path): train_list=os.listdir(train_img_path) train_label_0=re.compile("train\d+_0\.jpg") train_label_1=re.compile("train\d+_1\.jpg") train_label=[] for img_name in train_list: if re.fullmatch(train_label_0,img_name): train_label.append(0) if re.fullmatch(train_label_1,img_name): train_label.append(1) #print(train_label) return train_label def createTestLabels(test_img_path): test_list=os.listdir(test_img_path) test_label_0=re.compile("test\d+_0\.jpg") test_label_1=re.compile("test\d+_1\.jpg") test_label=[] for img_name in test_list: if re.fullmatch(test_label_0,img_name): test_label.append(0) if re.fullmatch(test_label_1,img_name): test_label.append(1) print("testlabel",len(test_label)) return test_label def readImages(img_path): img_list=os.listdir(img_path) i=0 IMG=np.zeros([28*28,len(img_list)],dtype=np.uint8) for img in img_list: img=cv2.imread(img_path+"\\"+img,0) img[img<127]=0 img[img>=127]=1 img.resize([28*28,1]) IMG[:,i]=img[:,0] i+=1 return IMG if __name__=="__main__": train_labels=createTrainLabels(r"E:\download_datasets\MNIST\0_1\0_1_train")# return a list test_labels=createTestLabels(r"E:\download_datasets\MNIST\0_1\0_1_test")# return a list train_images=readImages(r"E:\download_datasets\MNIST\0_1\0_1_train")# return an array test_images=readImages(r"E:\download_datasets\MNIST\0_1\0_1_test")# return an array train_number=len(train_labels) test_number=len(test_labels) features=28*28 X=train_images Y=(np.array(train_labels)).reshape([1,train_number]) r=0.01 W=np.random.random([1,features]) b=np.random.random([1,1]) for i in range(100): Z=np.dot(W,X)+b a=1/(1+np.exp(-Z)) dW=np.dot((a-Y),X.T) db=np.dot((a-Y),np.ones([train_number,features]))[0,0] #db=db_array W=W-r*dW b=b-r*db Z_t=np.dot(W,test_images)+b a_t=1/(1+np.exp(-Z_t)) a_t[a_t<0.5]=int(0) a_t[a_t>=0.5]=int(1) predict=a_t print("Predict:",predict) right=(np.array(test_labels)).reshape([1,test_number]) print("TestLabels:",right) errors=predict-right print("Errors:",errors)