基于knn的图片识别

发布时间 2023-11-07 10:59:50作者: 一枚码农

基于KNN的图片识别

KNN算法介绍

image
https://zhuanlan.zhihu.com/p/25994179

项目简介

使用numpy和opencv实现的一个简易的knn图片识别模块

项目结构

- code: 存放代码
  - create_collection.py: 创建数据集功能, 会将img中的图片建立数据集到data_collection下
  - knn.py: knn实现
- data_collection: 数据集存放目录
- img: 生成数据集的原始图片存放目录,生成数据集之后可以删掉
- test_data: 测试数据图片存放目录

使用

执行knn.py文件

测试结果展示

输入测试数据中的car.jpg(车)图片,k=3,执行knn.py之后,得到:
image

代码仓库地址

https://github.com/2267770481/knn/tree/master