基于HOG和SVM的图像识别方法
- 格式:pdf
- 大小:495.85 KB
- 文档页数:6
http://www.paper.edu.cn
- 1 - 中国科技论文在线
基于HOG和SVM的图像识别方法
林锦彬,庄伯金**
作者简介:林锦彬(1989-),男,硕士,主要研究方向:多媒体通信与模式识别、嵌入式系统
通信联系人:庄伯金(1976-),男,副教授,主要研究方向:嵌入式系统,MPEG-2/MPEG-4/H.263/H.264
智能转码器设计以及网络编码(Network Coding)与视频应用融合技术等. E-mail: bjzhuang@bupt.edu.cn (北京邮电大学信息与通信工程学院,北京 100876)
5 摘要:图像识别技术目前已广泛应用于人们的工作与生活中,不同的应用场景对图像识别技
术提出了不同的应用要求。本文提出一种基于HOG和SVM的图像识别方法。首先对输入
图像进行校正与归一化等预处理,去除背景噪声的干扰,然后利用梯度方向直方图(HOG)
提取图像特征,建立对图像的鲁棒表示,最后使用SVM分类器对送入的图像特征模式进行
分类。为了降低计算复杂度,采用了一种积分图方法对HOG特征提取过程进行了优化。实10
验结果表明,本文提出的方法能够较好地对图像进行分类,具备较高的识别准确率。
关键词:图像识别;投影变换;HOG;SVM;积分图
中图分类号:TP391.4
Image recognition method based on HOG and SVM 15
LIN Jinbin, ZHUANG Bojin
(Information and Communication Engineering School, Beijing University of Posts and
Telecommunications, Beijing 100876)
Abstract: Nowadays image recognition technology is widely employed in our work and daily life,
however different application scenarios put forward different requirements. An image recognition 20
method based on HOG and SVM is proposed. Firstly, in order to remove background noise, an image
correction and normalization is required. Secondly, we establish a robust representation of image by
using HOG feature extraction. Finally we classify different input feature vectors by using SVM. We
take an integral image approach, which is applied to optimize HOG feature extraction, to reduced the
computational complexity. The experiment results show that our method achieve a good performance 25
on image recognition.
Key words: Image Recognition; Projection Transformation; HOG; SVM; Integral Image
0 引言
图像识别是指利用计算机对图像进行处理和分析以识别不同模式的目标和对象的技术。30
借助图像识别技术,人们可以快速获得所需要的信息,从而更好地进行思考与决策。随着计
算机技术和信息技术的不断发展,图像识别得到了越来越广泛的应用。图像识别一般可以分
为三个部分,即图像预处理、特征提取与图像分类。图像预处理的主要目的是消除图像中无
关的信息,恢复有用的真实信息,一般有数字化、几何变换、归一化、滤波等步骤。特征提
取是目标识别的关键,特征的选择和描述是识别性能的直接体现,常用的图像特征有颜色特35
征、纹理特征、形状特征和空间关系特征等。颜色特征是一种全局特征,描述图像或图像区
域所对应的表现性质,比如灰度直方图等。纹理特征也是一种全局特征,描述图像或图像区
域的空间颜色分布和光强分布,比如LBP特征、灰度共生矩阵等。形状特征是一种局部特
征,描述了局部区域内物体的外形的性质,比如矩特征、边界特征等。空间关系特征是指图
像分割出来的多个目标之间相互的空间关系或相对方向关系,包括连接/邻接关系、交叠/重40
叠关系和包含/包容关系等[1]。图像分类技术根据输入的特征向量计算出它所属的类别,常见
的方法有模板匹配法、神经网络与支撑向量机等。 http://www.paper.edu.cn
- 2 - 中国科技论文在线
本文根据图像的特点,进行了校正与归一化等图像预处理。提出了基于HOG与SVM
的图像识别方法,并使用该算法对样本图像进行了测试,取得了满意的效果。
1 图像前景提取 45
本文中需要识别的目标图像为规则的矩形物体,经过图像采集后呈现倾斜、拉伸和
变形等情况,如图2(a)所示。本文采用一种基于图像投影变换模型的方法进行校正与归一化,
基本思想是计算出原始图像中的目标图像到指定矩形模板的投影变换矩阵,将原始目标图像
的所有像素映射到指定矩形模板内,从而实现目标图像的校正[2]。投影变换模型示意图如下:
50
(a) 原图 (b) 变换图
图1 投影变换示意图
投影变换矩阵为:
012
345
671mmm
Hmmm
mm
假设原图上任一点坐标为
11(,)xy,校正后图像对应点坐标为
22(,)xy,则有: 55
12
12
11xx
yHy
本文中将目标图像归一化到640×320大小,通过定位出原图中目标图像的4对顶点,
如图2(a)中的红色交叉点所示,校正后图像的4对对应参考点分别为(0,0),(639,0),(0,319),
(639,319),使用高斯消元法可以解出变换矩阵H的8个未知变量。对于点
22(,)xy在原始图
像中的映射坐标
11(,)xy,通过插值算法[3]即可以得到其对应的像素值,也即完成了整个图像60
校正过程。
经过校正及归一化后的图像如图2(b)所示。
(a) 原图输入图像 (b) 校正及归一化后图像
图2 图像校正与归一化效果图 65
2 HOG特征提取
HOG特征是一种在计算机视觉和模式识别中用来进行物体检测的特征描述子,由
Navneet Dalal和Bill Triggs首先在05年的CVPR中提出,并用于静态图像与视频的行人检 http://www.paper.edu.cn
- 3 - 中国科技论文在线
测[4]。HOG特征的核心思想是所检测的局部物体外形能够被光强梯度或边缘方向的分布所
描述,它通过计算和统计图像局部区域的梯度方向直方图来构成特征。具体步骤如下: 70
先将图像灰度化,采用Gamma校正法进行颜色空间的标准化,同时也能起到抑制噪声、
降低光照变化以及图像局部阴影的影响。公式为:
(,)(,)IxyIxyg
其中(,)Ixy 表示图像灰度值,通常取1/2g 。
计算像素点(,)xy的梯度: 75
(,)(1,)(1,)
xGxyIxyIxy
(,)(,1)(,1)
yGxyIxyIxy
式中,(,)xGxy、(,)
yGxy、(,)Ixy分别表示输入图像中像素点(,)xy处的水平方向梯度、
垂直方向梯度和像素值。像素点(,)xy处的梯度幅值和梯度方向分别为: 22(,)(,)(,)
xyGxyGxyGxy 80 (,)
(,)tan
(,)y
xGxy
xyarc
Gxyq
HOG特征将整幅图像分割成小的连接区域(cell),在每个cell内对所有像素的梯度方
向根据幅度大小进行加权投票,这样每个cell生成一个方向梯度直方图。本文设定cell大小
为8,梯度方向划分为16个区间。
把若干个cell组合成一个区间(block),在这个block内对所有cell的梯度强度进行归85
一化,这个过程使HOG特征具备更好的光照、阴影不变性。注意这些block之间是互有重
叠的。本文设定block大小为2×2个cell,则相邻block的重叠cell数为2。
将所有block的特征向量组合起来得到最终的HOG特征。
对于128×64的图像,按照上述的计算方法,则水平方向有15个扫描窗口,垂直方
向有7个扫描窗口,每个block内的特征向量维数为16×4=64,最终HOG特征的维数为90
15×7×64=6720维。
3 SVM分类器
对不同类别的图像提取HOG特征后,下一步就需要进行分类。本文采用的方法是
支撑向量机。
SVM是在统计学习理论的基础上发展起来的机器学习算法。上世纪60年代,Vapnik95
等人开始了统计学习理论的研究,并于上世纪60年代至70年代建立了统计学习理论的框架。
基于Vapnik的统计学习理论,针对小样本分类和回归问题的新型机器学习方法支撑向量机
被提了出来。支撑向量机以结构风险最小化原理为准则,能够有效解决过拟合、落入局部极
值和“维数灾难”等问题,在解决小样本、非线性、高维空间模式识别问题中表现出良好的推
广性和较好的分类准确性[5]。 100
采用SVM分类器进行面值识别的具体实现分为两个阶段:训练阶段和预测阶段。
训练阶段,使用训练样本进行学习并训练出一个SVM模型,这个模型本质上就是由最难分
类的特征向量定义出了最优超平面。预测阶段将待测样本特征向量输入到k(k-1)/2个二分类
器,通过投票数给出分类的结果。系统框图如下: