python svm算法 cifar_python实现HOG+SVM对CIFAR-10数据集分类(上)

本博客只用于学习,如果有错误的地方,恳请指正,如需转载请注明出处。

看机器学习也是有一段时间了,这两天终于勇敢地踏出了第一步,实现了HOG+SVM对图片分类,具体代码可以在github上下载,https://github.com/subicWang/HOG-SVM-classifer。大家都说HOG+SVM是在行人检测中很合拍的一对。至于为啥,我也讲不清楚。我猜想这么合拍的一对应用在图片分类上效果应该也不错吧,事实证明确实还行,速度挺快,分类正确率还行。我用的数据集是http://www.cs.toronto.edu/~kriz/cifar.html。图片特征HOG的提取过程,本文不做讲解,很多博客也肯定比我讲得清楚。那我就直接粘出我的代码吧,方便需要的人参考。

def getHOGfeat( image,stride = 8, orientations=8, pixels_per_cell=(8, 8),cells_per_block=(2, 2)):

cx, cy = pixels_per_cell

bx, by = cells_per_block

sx, sy = image.shape

n_cellsx = int(np.floor(sx // cx)) # number of cells in x

n_cellsy = int(np.floor(sy // cy)) # number of cells in y

n_blocksx = (n_cellsx - bx) + 1

n_blocksy = (n_cellsy - by) + 1

gx = zeros((sx, sy), dtype=np.d

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
要使用Python-OpenCV训练SVM,您需要遵循以下步骤: 1. 收集并准备数据集:您需要准备正样本和负样本的图像。将它们分别放在两个文件夹中,并确保它们的命名方式相同。 2. 提取图像特征:使用OpenCV的HOG描述符提取每个图像的特征。将这些特征存储在一个numpy数组中。 3. 创建SVM训练器:使用OpenCV的ml库创建一个SVM训练器对象。 4. 训练SVM:调用SVM训练器的train方法,并将特征数组和标签数组作为参数传递。标签数组应该是一个大小相同的numpy数组,其中包含正样本和负样本的标签。 5. 测试SVM:使用一些测试数据来测试SVM的性能,可以使用OpenCV的predict方法。预测结果将是一个数字,其中1表示正类,0表示负类。 以下是一个简单的示例代码,用于训练SVM: ```python import cv2 import numpy as np # Step 1: Collect and prepare dataset pos_images = ["pos1.jpg", "pos2.jpg", ...] neg_images = ["neg1.jpg", "neg2.jpg", ...] # Step 2: Extract image features hog = cv2.HOGDescriptor() features = [] labels = [] for img in pos_images: img = cv2.imread(img) features.append(hog.compute(img)) labels.append(1) for img in neg_images: img = cv2.imread(img) features.append(hog.compute(img)) labels.append(0) features = np.squeeze(features) labels = np.array(labels) # Step 3: Create SVM trainer svm = cv2.ml.SVM_create() # Step 4: Train SVM svm.train(features, cv2.ml.ROW_SAMPLE, labels) # Step 5: Test SVM test_img = cv2.imread("test.jpg") test_features = hog.compute(test_img) result = svm.predict(test_features) print(result) ``` 请注意,此代码仅为示例,您可能需要根据自己的数据集和需求进行修改。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值