机器学习算法与Python实践之 k近邻（KNN）

本文主要参考博客http://blog.csdn.net/zouxy09/article/details/16955347

，只是加入了自己的部分，测试数据是自己用windows自带的画图软件画出来

import cv2
import numpy as np
import math
from matplotlib import pyplot as plt

def save(filename, contents):
  fh = open(filename, 'a')
  fh.write(contents)
  fh.close()
filename = 'E:/西普教育/digits/testDigits/9_90.txt'
img = cv2.imread('E:/data/6.jpg', 0)
ret,thresh1 = cv2.threshold(img, 127, 255, cv2.THRESH_BINARY)
plt.imshow(thresh1,'gray')
plt.title([])
plt.xticks([]),plt.yticks([])
plt.show()
#print(img)
fh = open(filename, 'w')
fh.write('')
fh.close()
a=img.tolist()
for i in range(0,32):
    for j in range(0,32):
        b=a[i][j]
        save(filename,str(abs(math.ceil(b/255/2)-1)))
        #if j!=31:
           # save('E:/data/6.txt',' ')
    save(filename,'\n')