#得到分类数据和测试数据
import pymysql
import struct
from numpy import *
a=['']*20 #存图像 分类数据
b=[[0]*76800]*20#存图像 分类数据
c=[0]*76800#存图像 测试数据
def connectSql_1():
conn = pymysql.connect(host='192.168.9.163', user='hlyxtmi', passwd='19560530', db='tmi-ds', charset="latin1")
cur = conn.cursor()
cur.execute("select type,imagedata from knn_test where type != ''")
results = cur.fetchall()
i=0
for row in results:
d = [0] * 76800
a[i] = row[0]
imagedata = row[1]
for num in range(76800):
str = imagedata[(num+4)*2:(num+5)*2]
d[num] = struct.unpack('h', str)
b[i]=d
i+=1
cur.close()
conn.close()
def connectSql_2():
conn = pymysql.connect(host='192.168.9.163', user='hlyxtmi', passwd='19560530', db='tmi-ds', charset="latin1")
cur = conn.cursor()
cur.execute("select imagedata from knn_test where type = ''")
results = cur.fetchall()
for row in results:
imagedata = row[0]
for num in range(76800):
str = imagedata[(num+4)*2:(num+5)*2]
c[num] = struct.unpack('h', str)
cur.close()
conn.close()
def classify():
s=[0]*20 #存距离
for num in range(20):
for i in range(76800):
aTmp = b[num][i][0]
bTmp = c[i][0]
cTmp = aTmp - bTmp
s[num]+= abs(cTmp)
print(num)
sortedDisIndex = argsort(s)
classCount={}
for i in range(19):
voteLabel = a[sortedDisIndex[i]]
classCount[voteLabel] = classCount.get(voteLabel,0) + 1
maxCount = 0
for key,value in classCount.items():
if value > maxCount:
maxCount = value
classes = key
return classes
if __name__ == '__main__':
connectSql_1()
connectSql_2()
output = classify()
print("分类结果为：",output)
