python knn自我实践

最新推荐文章于 2025-09-08 15:51:51 发布

转载最新推荐文章于 2025-09-08 15:51:51 发布 · 72 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/gaara-zhang/p/9713820.html

文章标签：

#python #人工智能

本文详细介绍了使用KNN算法进行图像分类的过程，包括从数据库获取图像数据，将图像数据转换为可处理的数值形式，以及利用KNN算法进行分类。通过连接特定数据库，读取图像数据和类型，对数据进行预处理，并实现KNN分类算法。

#得到分类数据和测试数据
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)

转载于:https://www.cnblogs.com/gaara-zhang/p/9713820.html