一种基于图像显著性与svm的飞机目标检测方法

文档序号:10594600阅读:560来源:国知局
一种基于图像显著性与svm的飞机目标检测方法
【专利摘要】本发明涉及一种基于图像显著性与SVM的飞机目标检测方法,显著性区域检测中有两种视觉注意方式,分别是:自底向上(数据驱动)的注意方式、自顶向下(任务驱动)的注意方式。本发明先提取训练样本的方向梯度直方图(HOG)特征,训练支持向量机(SVM)分类器,然后使用一种基于剩余谱理论的自底向上的视觉注意模型进行显著性区域检测,快速提取可能存在目标的候选区域,然后提取候选区域的HOG特征,再使用训练好的SVM分类器对候选区域进行分类,完成目标检测。
【专利说明】
-种基于图像显著性与SVM的飞机目标检测方法
技术领域
[0001] 本发明属于计算机图像处理,设及飞机目标检测方法,具体设及一种基于图像显 著性与SVM的飞机目标检测方法。
【背景技术】
[0002] 可见光图像目标检测是目标检测领域重要的分支之一,在军事领域有极为重要的 应用。目前,在飞机检测方面,国内外的专家学者已经做了大量的研究工作。比较早的有基 于图像角点与边缘特征的飞机检测方法,通过角点和边缘所围成的外观形状来判断飞机的 位置。另外,由于飞机在图像中的独特形状,现有技术中出现过一种用于飞机检测的圆滤波 器检测方法;但是,该方法容易受到飞机尺寸的限制,对尺度不合适的飞机区域采用圆滤波 器并不能达到好的检测效果。而基于图像显著性的飞机检测方法,则主要利用人眼的视觉 系统特点初步获取图像中的飞机可疑区域,然后与目标特征相结合实现对飞机位置的检 测。
[0003] 在基于统计学习的方法中,可见光图像中的目标检测通常被认定为一个二分类问 题,即待检测目标要么是目标,要么不是目标。随着机器学习与特征提取等方法研究的不断 深入W及不断被应用到种类目标检测中,现有的机器学习方法与特征提取方法使得目标检 测的准确性有了很大的提高,但仍不能高效、准确地提取出图像中包含目标的候选区域。

【发明内容】

[0004] 要解决的技术问题
[0005] 为了避免现有技术的不足之处,本发明提出一种基于图像显著性与SVM的飞机目 标检测方法,将基于显著性区域检测的图像分割运用到目标检测领域W解决检测精度不高 等问题。
[0006] 技术方案
[0007] -种基于图像显著性与SVM的飞机目标检测方法,其特征在于步骤如下:
[000引步骤1、数据准备:从公众可获得的各类图像中选取可见光机场图像作为训练样本 及待检测目标图集,并转换为灰度图,图中飞机的最大外接正方形的尺寸从最小值:aiXai, 至最大值为32X32;选取部分图像作为训练样本,将图像中飞机区域切割后作为正样本,其 他区域切割后作为负样本;剩余的部分图像作为测试样本;所述的训练的样本缩放为33X33 大小的尺寸;
[0009] 步骤2:分别提取正负训练样本的HOG特征;
[0010] 步骤3:将提取到的每一个正负样本的HOG特征与其类别标签,正样本为1,负样本 为-1组合成一个向量,训练SVM分类器;
[0011] 步骤4:采用基于剩余谱理论的显著性检测方法获取测试样本中图像的显著图;
[0012] 步骤5:通过面积阔值化方法提取显著图中的连通域,产生飞机目标的候选区域, 过程如下:
[0014]
[oou]步骤5曰:首先计算显著图的均值另和方差沪;
[0015]
[0016] 其中,W表示图像的宽,h表示图像的高,SaKiJ)表示显著图中第i行第j列的像素 值;
[0017] 步骤化:利用显著图的均值^和方差系计算阔值T,W该阔值T分割显著图;所述阔 值T:
[001 引
[0019] 系数k是为了平衡标准差.容和均值//在阔值T取值时的权重A = ex-p(.-(V/15);
[0020] 步骤5c、对阔值化后的显著图进行连通域滤波:W大小为mXm的矩形窗,W第i行 第j列像素Q J)为中屯、截取一个邻域;W某区域的像素个数来定义该区域的面积,求出处 于所述邻域范围内的显著区域的面积;如果该面积大于一定阔值,就保留该区域,否则不保 留;将所有保留下来的区域作为飞机目标的候选区域;
[0021 ]步骤6:从飞机目标的候选区域提取飞机目标,过程如下:
[0022] 步骤6a、构建响应矩阵:
[0023] 构建一个二维的响应矩阵M,矩阵中每个元素的取值为一个二维数组Mij(s,r)其 中i G [1 ,W] ,j G [1 ,h],s用于记录窗口尺寸,r用于记录存在飞机目标的窗口所产生的SVM 响应;初始化时,该响应矩阵初始化为一个与图像大小相同的全0矩阵;
[0024] W任一候选区域的局部像素极大值作为该候选区域的中屯、,称为:种子点;
[0025] 步骤化、二次窗口法剔除无效种子点,保留最可能的飞机区域,过程如下:
[00%] (A)确定第一次图像块窗口 :窗口是一个W种子点为中屯、的矩形窗口,窗口的尺寸 为hiXhi,其中:
[0027]
[00%]其中,into函数表示四舍五入取整。W该窗口对原图像进行截取,将截取到的图 像块缩放为33X33大小,提取图像块的HOG特征,然后将HOG特征代入到SVM分类器中,计算 SVM分类器响应值ri;
[0029] (B)确定第二次图像块窗口 :窗口是一个W种子点为中屯、的矩形窗口,窗口的尺寸 为h2Xh2,其中:
[0030] h2 = 2(hi-l)+l
[0031] W该窗口对原图像进行截取,将截取到的图像块缩放为a3Xa3大小,提取图像块的 HOG特征,然后将HOG特征代入到SVM分类器中,计算SVM分类器响应值K;
[0032] (C)判断是否保留该区域:
[0033] 如果两次计算得到的分类器响应值均小于0,则认为该种子点对应的飞机候选区 域无效,因此剔除该无效种子点;
[0034] 否则,认为窗口包含的区域存在飞机或包含了飞机的大部分机身,将该两次响应 中的较大值赋值到响应矩阵中种子点对应的坐标位置上,称为响应矩阵的种子点,并记录 下该响应对应的窗口尺寸;具体地,设当前坐标为:(u,v),将需要记录的SVM分类器响应值, 赋给响应矩阵的种子点处元素M(u,v)的二维数组中的r值;将产生该响应时对应的截取窗 口尺寸中的正方形边长值,赋给响应矩阵的种子点处元素M(u,v)的二维数组中的S值;
[0035] 步骤6c、优化响应矩阵,确定飞机区域:
[0036] (1)优化每个种子点中记录的窗口尺寸:
[0037] 对每一个响应矩阵的种子点,先根据W下公式计算变换后的窗口边长值h3:
[00;3 引
[0039] W响应矩阵的种子点处的坐标为中屯、,Wh3Xh3为窗口尺寸建立截取窗口对原图 像进行截取,将截取到的图像块缩放为曰3 X曰3大小,提取图像块的HOG特征,然后将HOG特征 代入到SVM分类器中,计算SVM分类器响应值n;
[0040] 再根据W下公式计算变换后的窗口边长值h4:
[0041]
[0042] W响应矩阵的种子点处的坐标为中屯、,Wh4 X h4为窗口尺寸建立截取窗口对原图 像进行截取,将截取到的图像块缩放为曰3 X曰3大小,提取图像块的HOG特征,然后将HOG特征 代入到SVM分类器中,计算SVM分类器响应值r4;
[0043] 找出r,r3,r4中的最大值,替换响应矩阵的种子点处元素的二维数组中原有的HI, W所述新的r值对应的窗口边长值替换响应矩阵的种子点处元素的二维数组中原有的S值;
[0044] 确定飞机区域,得到最终目标检测结果;
[0045] (2)依据优化后的响应矩阵,W响应矩阵的种子点处记录的坐标作为原图像中飞 机目标所在正方形区域的中屯、,W该种子点处记录的S值作为原图像中飞机目标所在正方 形区域的边长,即可确定每个飞机区域,得到最终的飞机目标检测结果。
[0046] 有益效果
[0047] 本发明提出的一种基于图像显著性与SVM的飞机目标检测方法,显著性区域检测 中有两种视觉注意方式,分别是:自底向上(数据驱动)的注意方式、自顶向下(任务驱动)的 注意方式。本发明先提取训练样本的方向梯度直方图化OG)特征,训练支持向量机(SVM)分 类器,然后使用一种基于剩余谱理论的自底向上的视觉注意模型进行显著性区域检测,快 速提取可能存在目标的候选区域,然后提取候选区域的HOG特征,再使用训练好的SVM分类 器对候选区域进行分类,完成目标检测。
[004引本发明将基于剩余谱理论的图像显著性检测与支持向量机的学习方法巧妙地结 合起来,改进了目标区域的选择,提高了检测精度和检测的鲁棒性,模型简单,执行效率高。
【附图说明】
[0049] 图1:基于SVM的检测网络的训练流程图
[0050] 图2:基于图像显著性与SVM的可见光图像飞机检测流程图
【具体实施方式】
[0051] 现结合实施例、附图对本发明作进一步描述:
[0052] 实施方案步骤如下:
[0化3]步骤1:数据准备。
[0054] 从G00GLE-EARTH上截取可见光机场图像作为训练样本及待检测目标,统一转换为 灰度图,图中飞机的最大外接正方形的尺寸从最小值:曰1 Xai,至最大值为曰2 X曰2不等。在本 实施例中,山=20,曰2 = 80。从运些截取后的图像中,采集用于训练的正负样本(各2000个), 人工标注其飞机区域。正样本为各种姿态、不同型号的飞机,负样本为飞机场中存在的建筑 物、飞机场内的标示、飞机尾部、飞机场中道路的交叉口等。在训练时所用样本均被统一缩 放为33X33大小的尺寸,对于每幅测试图像,不需要裁剪。在本实施例中,曰3 = 64。
[0055] 步骤2:提取正负训练样本的HOG特征。
[0056] 先使用Gamma校正的方法对缩放后尺寸为a3Xa3的图像归一化。然后将图像均匀地 划分成bi Xbi个"单元格"(cell),每个单元格的大小为Cl Xci个像素,其中:ci = a3/bi。为便 于计算,通常将a3,bi,ci的取值调整为整数值,本实施例中,bi = 8,ci = 8。求出单元格内每个 像素的梯度方向,根据其梯度方向的幅值加权投影到对应的直方图区间中,一个单元格对 应一个N维(运里N=9,即从每个单元格中提取9维特征)的向量。随后取b2 X b2个单元格组合 成一个块(block),块与块的起始位置之间的间隔取di个像素,使得块之间互有重叠,块的 选取顺序为:先从左到右,再从上到下。在本实施例中,b2 = 2,di = 8,即:水平方向和垂直方 向分别有7个块,使得图像共包含49个块。每个单元格仍取N维特征,块的特征由单元格的直 方图特征向量按照单元格的排列顺序串联组成;整个图像的册G特征则由各个块的册G特征 向量串联组成。在本实施例中,每个块有9 X 2 X 2共36维特征,整个归一化图像共有36 X 49 共1764维特征。
[0057] 步骤3:将上一步中提取到的每一个正负样本的HOG特征与其类别标签(正样本为 1,负样本为-1)组合成一个向量,训练SVM分类器。本实施例中SVM的核函数采用径向基核函 数。
[0058] 步骤4:训练完毕后,使用基于剩余谱理论的显著性检测方法获取测试图像的显著 图。
[0059] (1)设待检测灰度图像I的宽度为W个像素,高度为h个像素,对图像的像素
[0060] 集合I(X)(其中I(x)G[0,255],xG[l,w*h])进行傅里叶变换F(Kx)),并提取图 像频
[0061 ]域的相位特征与振幅特征;
[0062] A(f)=R(F(I(x))) (1)
[0063] P(f)=S(F(I(x))) (2)
[0064] 其中,A(f)表示频率f的振幅,P(f)表示频率f的相位,R(FQU)))表示取F(Kx)) 振幅,S(FQU)))表示取F(Kx))相位。
[0065] (2)计算振幅的log谱Uf ),并对log谱进行均值滤波,然后求log谱的剩余谱R(f):
[0066] L(f) = log(A(f)) (3)
[0067] R(f)=L(f)-hn(f)A(f) (4)
[006引上式中hn(f)是大小为n Xn(本实施方案中选择3 X 3大小)均值滤波卷积核,定义 为
[0069]
[0070] (3)将剩余谱与相位谱结合求得新的图像频谱,对该图像频谱进行傅里叶反变换 求得新的图像,并对得到的图像利用高斯卷积核进行图像平滑,即可得到基于剩余谱理论 的图像显著图S(X):
[0071] S(x)=g(x)冲-i[exp(R(f)+P(f))]2 (5)
[0072] 上式中,Fi(f)为傅里叶反变换,g(x)表示高斯卷积核。
[0073] 步骤5:通过面积阔值化方法提取显著图中的连通域,W此产生飞机目标的候选区 域。
[0074] (1)首先计算显著图的均值和方差沪.
[0075] (6)
[0076] 饥
[0077] 其中,W表示图像的宽,h表示图像的高,SaKiJ)表示显著图中第i行第j列的像素 值。
[0078] (2)利用显著图的均值和方差沪计算阔值T,W该阔值T分割显著图。运里的阔值 T:
[0079] (8)
[0080] 系数k是为了平衡标准差^和均值另在阔值T取值时的权重,本实施例中: Z、二exp(--15) 运样取值可W使阔值T与标准差多满足一定的反比特性,降低标准差过大 时对阔值的影响,或在标准差过小的时候能适当的增加标准差的影响。
[0081] (3)对阔值化后的显著图进行连通域滤波。W大小为m Xm的矩形窗,W第i行第j 列像素(ij)为中屯、截取一个邻域,本实施例中,IU = IIdW某区域的像素个数来定义该区 域的面积,可求出处于所述邻域范围内的显著区域的面积。如果该面积大于一定阔值(本实 施例中,阔值取60),就保留该区域,否则不保留。将所有保留下来的区域作为飞机目标的候 选区域。
[0082] 步骤6:从飞机目标的候选区域提取飞机目标。
[0083] (1)构建响应矩阵
[0084] 构建一个二维的响应矩阵M,矩阵中每个元素的取值为一个二维数组Mi, j(S,r)其 中i G [1,w],j G [1,h],s用于记录窗口尺寸,!用于记录存在飞机目标的窗口所产生的SVM 响应。初始化时,该响应矩阵初始化为一个与图像大小相同的全0矩阵。
[0085] W某个候选区域的局部像素极大值作为该候选区域的中屯、,称为:种子点(即:可 能存在飞机目标的区域中屯、位置)。
[0086] (2)剔除无效种子点,保留最可能的飞机区域
[0087] 为提高筛选的准确率,本发明提出一种二次窗口法,具体如下:
[0088] (A)确定第一次图像块窗口
[0089] 该窗口是一个W种子点为中屯、的矩形窗口,窗口的尺寸为hiXhi,其中:
[0090]
巧)
[0091] 其中,into函数表示四舍五入取整。W该窗口对原图像进行截取,将截取到的图 像块缩放为33X33大小,提取图像块的HOG特征,然后将HOG特征代入到SVM分类器中,计算 SVM分类器响应值ri。
[0092] (B)确定第二次图像块窗口
[0093] 该窗口是一个W种子点为中屯、的矩形窗口,窗口的尺寸为h2Xh2,其中:
[0094] h2 = 2(hi-l)+l (10)
[00%] W该窗口对原图像进行截取,将截取到的图像块缩放为a3Xa3大小,提取图像块的 HOG特征,然后将HOG特征代入到SVM分类器中,计算SVM分类器响应值K。
[0096] (C)判断是否保留该区域
[0097] 如果两次计算得到的分类器响应值均小于0,则认为该种子点对应的飞机候选区 域无效,因此剔除该无效种子点。
[0098] 否则,认为窗口包含的区域存在飞机或包含了飞机的大部分机身,可将该两次响 应中的较大值赋值到所述响应矩阵中种子点对应的坐标位置上,该点称为响应矩阵的种子 点,并记录下该响应对应的窗口尺寸。具体地,设当前坐标为:(u,v),将需要记录的SVM分类 器响应值,赋给响应矩阵的种子点处元素M(u,V)的二维数组中的r值;将产生该响应时对应 的截取窗口尺寸中的正方形边长值,赋给响应矩阵的种子点处元素M(u,v)的二维数组中的 S值。
[0099] (3)优化响应矩阵,确定飞机区域
[0100] (A)优化每个种子点中记录的窗口尺寸
[0101] 对毎一个晌巧巧随的种子点,先根据W下公式计算变换后的窗口边长值h3:
[0102]
(U)
[0103] W响应矩阵的种子点处的坐标为中屯、,Wh3Xh3为窗口尺寸建立截取窗口对原图 像进行截取,将截取到的图像块缩放为曰3 X曰3大小,提取图像块的HOG特征,然后将HOG特征 代入到SVM分类器中,计算SVM分类器响应值n。
[0104] 再根据W下公式计算变换后的窗口边长值h4:
[010 引
(12)
[0106] W响应矩阵的种子点处的坐标为中屯、,Wh4Xh4为窗口尺寸建立截取窗口对原图 像进行截取,将截取到的图像块缩放为曰3 X曰3大小,提取图像块的HOG特征,然后将HOG特征 代入到SVM分类器中,计算SVM分类器响应值r4。
[0107] 找出r,n,T4中的最大值,替换响应矩阵的种子点处元素的二维数组中原有的r值, W所述新的r值对应的窗口边长值替换响应矩阵的种子点处元素的二维数组中原有的S值。
[0108] (B)确定飞机区域,得到最终目标检测结果
[0109]依据优化后的响应矩阵,W响应矩阵的种子点处记录的坐标作为原图像中飞机目 标所在正方形区域的中屯、,W该种子点处记录的S值作为原图像中飞机目标所在正方形区 域的边长,即可确定每个飞机区域,得到最终的飞机目标检测结果。
【主权项】
1. 一种基于图像显著性与SVM的飞机目标检测方法,其特征在于步骤如下: 步骤1、数据准备:从公众可获得的各类图像中选取可见光机场图像作为训练样本及待 检测目标图集,并转换为灰度图,图中飞机的最大外接正方形的尺寸从最小值:ai Xai,至最 大值为a2Xa2;选取部分图像作为训练样本,将图像中飞机区域切割后作为正样本,其他区 域切割后作为负样本;剩余的部分图像作为测试样本;所述的训练的样本缩放为a 3Xa3大小 的尺寸; 步骤2:分别提取正负训练样本的HOG特征; 步骤3:将提取到的每一个正负样本的HOG特征与其类别标签,正样本为1,负样本为-1 组合成一个向量,训练SVM分类器; 步骤4:采用基于剩余谱理论的显著性检测方法获取测试样本中图像的显著图; 步骤5:通过面积阈值化方法提取显著图中的连通域,产生飞机目标的候选区域,过程 如下: 步骤5a:首先计算显著图的均值&和方差7: 其中,w表示图像的宽,h表示图像的高,Sal(i,j)表示显著图中第i行第j列的像素值; 步骤5b:利用显著图的均值i和方差歹计算阈值T,以该阈值T分割显著图;所述阈值T:系数k是为了平衡标准差表和均值;?在阈值T取值时的权重A = expC-i715); 步骤5c、对阈值化后的显著图进行连通域滤波:以大小为mXm的矩形窗,以第i行第j 列像素(i,j)为中心截取一个邻域;以某区域的像素个数来定义该区域的面积,求出处于所 述邻域范围内的显著区域的面积;如果该面积大于一定阈值,就保留该区域,否则不保留; 将所有保留下来的区域作为飞机目标的候选区域; 步骤6:从飞机目标的候选区域提取飞机目标,过程如下: 步骤6a、构建响应矩阵: 构建一个二维的响应矩阵M,矩阵中每个元素的取值为一个二维数组Mi, j(s,r)其中ie [1,w],je [1,h],s用于记录窗口尺寸,r用于记录存在飞机目标的窗口所产生的SVM响应; 初始化时,该响应矩阵初始化为一个与图像大小相同的全〇矩阵; 以任一候选区域的局部像素极大值作为该候选区域的中心,称为:种子点; 步骤6b、二次窗口法剔除无效种子点,保留最可能的飞机区域,过程如下: (A)确定第一次图像块窗口:窗口是一个以种子点为中心的矩形窗口,窗口的尺寸为In Xlu,其中:其中,int()函数表示四舍五入取整。以该窗口对原图像进行截取,将截取到的图像块 缩放为a3 Xa3大小,提取图像块的HOG特征,然后将HOG特征代入到SVM分类器中,计算SVM分 类器响应值η; (Β)确定第二次图像块窗口 :窗口是一个以种子点为中心的矩形窗口,窗口的尺寸为^ Xh2,其中: h2 = 2(hi_l)+l 以该窗口对原图像进行截取,将截取到的图像块缩放为a3Xa3大小,提取图像块的HOG 特征,然后将HOG特征代入到SVM分类器中,计算SVM分类器响应值r2; (C)判断是否保留该区域: 如果两次计算得到的分类器响应值均小于0,则认为该种子点对应的飞机候选区域无 效,因此剔除该无效种子点; 否则,认为窗口包含的区域存在飞机或包含了飞机的大部分机身,将该两次响应中的 较大值赋值到响应矩阵中种子点对应的坐标位置上,称为响应矩阵的种子点,并记录下该 响应对应的窗口尺寸;具体地,设当前坐标为:(u,v),将需要记录的SVM分类器响应值,赋给 响应矩阵的种子点处元素M(u,v)的二维数组中的r值;将产生该响应时对应的截取窗口尺 寸中的正方形边长值,赋给响应矩阵的种子点处元素M(u,v)的二维数组中的s值; 步骤6c、优化响应矩阵,确定飞机区域: (1) 优化每个种子点中记录的窗口尺寸: 对每一个响应矩阵的种子点,先根据以下公式计算变换后的窗口边长值h3:以响应矩阵的种子点处的坐标为中心,以h3 Xh3为窗口尺寸建立截取窗口对原图像进 行截取,将截取到的图像块缩放为a3 Xa3大小,提取图像块的HOG特征,然后将HOG特征代入 到SVM分类器中,计算SVM分类器响应值r 3; 再根据以下公式计算变换后的窗口边长值h4:以响应矩阵的种子点处的坐标为中心,以h4Xh4为窗口尺寸建立截取窗口对原图像进 行截取,将截取到的图像块缩放为a3 Xa3大小,提取图像块的HOG特征,然后将HOG特征代入 到SVM分类器中,计算SVM分类器响应值r4; 找出r,r3,r4中的最大值,替换响应矩阵的种子点处元素的二维数组中原有的r值,以所 述新的r值对应的窗口边长值替换响应矩阵的种子点处元素的二维数组中原有的s值; 确定飞机区域,得到最终目标检测结果; (2) 依据优化后的响应矩阵,以响应矩阵的种子点处记录的坐标作为原图像中飞机目 标所在正方形区域的中心,以该种子点处记录的s值作为原图像中飞机目标所在正方形区 域的边长,即可确定每个飞机区域,得到最终的飞机目标检测结果。
【文档编号】G06K9/46GK105956592SQ201610303628
【公开日】2016年9月21日
【申请日】2016年5月10日
【发明人】李映, 聂金苗, 陈迪
【申请人】西北工业大学
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1