基于python的hog+svm实现目标检测

本次是基于python的hog+svm实现目标检测,是对草莓的识别,大家可以自行更换数据集,代码简单移动,逻辑清晰,演示过程可以观看bilibili

上效果图: 

基于python的hog+svm实现目标检测_第1张图片

 本次的所有demo

基于python的hog+svm实现目标检测_第2张图片

算法部分

本次是基于python的hog+svm实现目标检测,是对草莓的识别。

数据集放在“data”文件夹下

“data”下的子文件夹有几个子文件夹就是几分类。

基于python的hog+svm实现目标检测_第3张图片

pos_img下放的都是草莓的照片,大家需要检测其他内容的可自行更换

 基于python的hog+svm实现目标检测_第4张图片

 neg_img文件夹下放的就是其他图片

代码依次运行

python 01svm_train.py

会在weights文件夹生成模型

python 02detect_photo.py

会调用模型,通过hog+nms对单张图片实现目标检测

基于python的hog+svm实现目标检测_第5张图片

 

 

你可能感兴趣的:(目标检测,人工智能,计算机视觉)