SVM 图片分类python实现

---

前言

SVM分类器就不介绍了，代码是网友通过github共享下载到的，源链接大家可以自行搜索一下。名称是SVMImageClassification-master。

---

一、数据集准备

每个文件夹一个图片类型。

二、安装环境

执行pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple ，后面加上清华源地址下载快一些。

三、执行代码

代码有几处需要修改。

第一处，文件夹名称和类别数量。
第二处，标签类别。

然后执行

python  svm.py

效果如下所示：

代码除了支持SVM，还支持KNN、决策树、朴素贝叶斯几种分类方法。只要取消注释即可。

代码使用颜色直方图的方式实现分类。
还可以使用HOG的方式实现分类。有需要可以私信我获取。

---

总结