使用介绍:

本项目为本人课程作业,使用SVM+HOG特征来实现螺钉和螺母的分类。

🎂食用方法:

​ 1.安装requirements.txt中的依赖

​ 2.进行训练,将test.py中的尾部代码do_training('images/train/pos', 30, 'images/train/neg', 30)取消注释, 运行test.py 即可。训练结束后,训练所得权重会保存在weights.txt中。

​ 3.测试,运行UI.py,点击选择图片,可以选择本文件夹images\test文件夹下的图片,也可以从网络自行下载测试图片。点击 进行检测

👀数据集介绍:

​ 数据集采用手机拍摄螺钉,螺母照片,分为正负样本。正样本为螺钉,负样本为螺帽。

❤友情提示:

​ 本项目中weights.txt,是我训练好的权重,可以直接使用。 也就是说,你不需要训练了,直接运行UI进行预测即可!!!!!

🐱‍👤运行结果:

image