基于Opencv的LBP+SVM分类实例

           LBP指局部二值模式,英文全称:Local Binary Pattern,是一种用来描述图像局部特征的算子,LBP特征具有灰度不变性和旋转不变性等显著优点。它是由T. Ojala, M.Pietikäinen, 和 D. Harwood [1][2]在1994年提出,由于LBP特征计算简单、效果较好,因此LBP特征在计算机视觉的许多领域都得到了广泛的应用,LBP特征比较出名的应用是用在人脸识别和目标检测中。

      LBP的种类(参考:http://blog.csdn.net/zouxy09/article/details/7929531,http://blog.csdn.net/quincuntial/article/details/50541815):

(1) 原始的LBP算子

原始的LBP算子定义为在3*3的窗口内,以窗口中心像素为阈值,将相邻的8个像素的灰度值与其进行比较,若周围像素值大于中心像素值,则该像素点的位置被标记为1,否则为0。这样,3*3邻域内的8个点经比较可产生8位二进制数(通常转换为十进制数即LBP码,共256种),即得到该窗口中心像素点的LBP值,并用这个值来反映该区域的纹理信息。



(2)圆形LBP算子

基本的 LBP算子的最大缺陷在于它只覆盖了一个固定半径范围内的小区域,这显然不能满足不同尺寸和频率纹理的需要。为了适应不同尺度的纹理特征,并达到灰度和旋转不变性的要求,Ojala等对 LBP 算子进行了改进,将 3×3邻域扩展到任意邻域,并用圆形邻域代替了正方形邻域,改进后的 LBP 算子允许在半径为 R 的圆形邻域内有任意多个像素点。从而得到了诸如半径为R的圆形区域内含有P个采样点的LBP算子。

(3)LBP旋转不变模式

 LBP 的定义可以看出,LBP 算子是灰度不变的,但却不是旋转不变的。图像的旋转就会得到不同的 LBP值。Maenpaa等人又将 LBP算子进行了扩展,提出了具有旋转不变性的 LBP 算子,即不断旋转圆形邻域得到一系列初始定义的 LBP值,取其最小值作为该邻域的 LBP 值。

(4)LBP等价模式

 Uniform Pattern,也被称为等价模式或均匀模式。一个LBP算子可以产生不同的二进制模式,对于半径为R的圆形区域内含有P个采样点的LBP算子将会产生P2种模式。很显然,随着邻域集内采样点数的增加,二进制模式的种类是急剧增加的。例如:5×5邻域内20个采样点,有2201,048,576种二进制模式。为了解决二进制模式过多的问题,提高统计性,Ojala提出了采用一种“等价模式”(Uniform Pattern)来对LBP算子的模式种类进行降维。Ojala等认为,在实际图像中,绝大多数LBP模式最多只包含两次从10或从01的跳变。因此,Ojala将“等价模式”定义为:当某个LBP所对应的循环二进制数从01或从10最多有两次跳变时,该LBP所对应的二进制就称为一个等价模式类。000000000次跳变),00000111(只含一次从01的跳变),10001111(先由1跳到0,再由0跳到1,共两次跳变)都是等价模式类。除等价模式类以外的模式都归为另一类,称为混合模式类,例如10010111(共四次跳变)。通过这样的改进,二进制模式的种类大大减少,而不会丢失任何信息。模式数量由原来的2P种减少为 P ( P-1)+2种,其中P表示邻域集内的采样点数。对于3×3邻域内8个采样点来说,二进制模式由原始的256种减少为58种,这使得特征向量的维数更少。

(5)MB-LBP特征

MB-LBP特征,全称为Multiscale Block LBP,中科院的人发明的,在Traincascade级联目标训练检测中的LBP特征使用的就是MB-LBP。将图像分成一个个小块(Block),每个小块再分为一个个的小区域(类似于HOG中的cell),小区域内的灰度平均值作为当前小区域的灰度值,与周围小区域灰度进行比较形成LBP特征,生成的特征称为MB-LBP,Block大小为3*3,则小区域的大小为1,就是原始的LBP特征,上图的Block大小为9*9,小区域的大小为3*3。

 

LBP特征向量进行提取的步骤

1)首先将检测窗口划分为16×16的小区域(cell);

2)对于每个cell中的一个像素,将相邻的8个像素的灰度值与其进行比较,若周围像素值大于中心像素值,则该像素点的位置被标记为1,否则为0。这样,3*3邻域内的8个点经比较可产生8位二进制数,即得到该窗口中心像素点的LBP值;

3)然后计算每个cell的直方图,即每个数字(假定是十进制数LBP值)出现的频率;然后对该直方图进行归一化处理。

4)最后将得到的每个cell的统计直方图进行连接成为一个特征向量,也就是整幅图的LBP纹理特征向量;

然后便可利用SVM或者其他机器学习算法进行分类了。参考上述博客中的代码实现效果如下:


这是opencv svm图像分类的整个工程代码,在VS2010下打开即可。整个工程文件以及我的所有训练的图片存放在这里,需要的可以下载,自己在找训练图片写代码花了很多时间,下载完后自行解压,训练图片和测试图片可以从这免费下载http://download.csdn.net/detail/always2015/8944959,project data文件夹直接放在D盘就行,里面存放训练的图片和待测试图片,以及训练过程中生成的中间文件,现在这个下载object_classfication_end则是工程文件,我用的是vs2010打开即可,下面工程里有几个要注意的地方: 1、在这个模块中使用到了c++的boost库,但是在这里有一个版本的限制。这个模块的代码只能在boost版本1.46以上使用,这个版本以下的就不能用了,直接运行就会出错,这是最需要注意的。因为在1.46版本以上中对比CsSVM这个类一些成员函数做了一些私有化的修改,所以在使用该类初始化对象时候需要注意。 2、我的模块所使用到的函数和产生的中间结果都是在一个categorizer类中声明的,由于不同的执行阶段中间结果有很多个,例如:训练图片聚类后所得到单词表矩阵,svm分类器的训练的结果等,中间结果的产生是相当耗时的,所以在刚开始就考虑到第一次运行时候把他以文件XML的格式保存下来,下次使用到的时候在读取。将一个矩阵存入文本的时候可以直接用输出流的方式将一个矩阵存入,但是读取时候如果用输入流直接一个矩阵变量的形式读取,那就肯定报错,因为输入流不支持直接对矩阵的操作,所以这时候只能对矩阵的元素一个一个进行读取了。 3、在测试的时候,如果输入的图片太小,或者全为黑色,当经过特征提取和单词构造完成使用svm进行分类时候会出现错误。经过调试代码,发现上述图片在生成该图片的单词的时候所得到的单词矩阵会是一个空矩阵,即该矩阵的行列数都为0,所以在使用svm分类器时候就出错。所以在使用每个输入图片的单词矩阵的时候先做一个判断,如果该矩阵行列数都为0,那么该图片直接跳过。
OpenCV中实现图像分类SVM(支持向量机)是一种基于机器学习的算法,它需要使用一组已经标记的图像来训练模型,然后使用该模型对未知图像进行分类。 以下是实现SVM进行图像分类的基本步骤: 1. 收集数据集,包括用于训练和测试的图像。 2. 对每个图像进行特征提取,例如计算每个图像的颜色直方图、纹理、边缘等特征。 3. 将特征向量作为输入,将训练图像标记为相应的类别。训练数据集的样本应该在同一类别内相似,而不同类别之间应该有显著差异。 4. 使用OpenCV中的SVM模块对训练数据集进行训练,并调整模型参数以提高准确性。 5. 对测试数据集中的每个图像进行特征提取,并使用训练好的模型进行分类。 6. 计算分类准确率和召回率,并根据需求调整参数,以提高模型的性能。 下面是示例代码来实现基于OpenCVSVM进行图像分类。 ``` python import cv2 import numpy as np # 读取数据集图像及其标签 train_images = [] train_labels = [] test_images = [] test_labels = [] for i in range(1, 5): for j in range(1, 6): img = cv2.imread("dataset/%d/%d.png" % (i, j), cv2.IMREAD_GRAYSCALE) # 读取灰度图像 if j < 4: # 训练集前3张图像 train_images.append(img) train_labels.append(i) else: # 测试集后2张图像 test_images.append(img) test_labels.append(i) # 提取图像特征 hog = cv2.HOGDescriptor() train_features = np.array([hog.compute(img) for img in train_images], dtype=np.float32) test_features = np.array([hog.compute(img) for img in test_images], dtype=np.float32) # 训练SVM分类svm = cv2.ml.SVM_create() svm.setType(cv2.ml.SVM_C_SVC) svm.setKernel(cv2.ml.SVM_LINEAR) svm.setTermCriteria((cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 1000, 0.01)) svm.train(train_features, cv2.ml.ROW_SAMPLE, np.array(train_labels)) # 测试SVM分类器 ret, pred = svm.predict(test_features) # 计算分类准确率和召回率 accuracy = np.mean(pred.squeeze() == np.array(test_labels)) recall = np.mean(pred.squeeze() == np.array(test_labels)[np.array(test_labels) == 1]) print("Accuracy: %.2f%%" % (accuracy * 100)) print("Recall: %.2f%%" % (recall * 100)) ``` 该示例代码演示了如何使用OpenCVSVM实现图像分类。它使用了一个简单的数据集,由四个类别的图像(数字1-4)构成。图像特征采用HOG描述符,SVM类型为C-SVC,核函数为线性核,迭代停止条件为1000次或0.01的精度。在测试过程中,计算了分类准确率和召回率两个指标,以评估分类器的性能。
评论 12
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值