作者:过去无法回去 | 来源:互联网 | 2023-09-25 14:14
前言
SVM分类器就不介绍了,代码是网友通过github共享下载到的,源链接大家可以自行搜索一下。名称是SVMImageClassification-master。
一、数据集准备
每个文件夹一个图片类型。
二、安装环境
执行pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
,后面加上清华源地址下载快一些。
三、执行代码
代码有几处需要修改。
第一处,文件夹名称和类别数量。
第二处,标签类别。
然后执行
python svm.py
效果如下所示:
代码除了支持SVM,还支持KNN、决策树、朴素贝叶斯几种分类方法。只要取消注释即可。
代码使用颜色直方图的方式实现分类。
还可以使用HOG的方式实现分类。有需要可以私信我获取。
总结