基于可变部件模型的电动摩托车图像检测方法技术

本发明专利技术公开了一种基于可变部件模型的电动摩托车图像检测方法，首先建立样本集，并进行特征提取，通过特征训练得到根模板，以空间聚合的方式获得初始根模板，对样本建立特征金字塔，训练所述的初始根模板；初始化部件模型并加载到初始根模板中经过训练得到最终根模板以及最终部件模型；对于待检测图像，对待检测图像进行特征提取，然后与所述的最终根模板以及最终部件模型进行匹配，得到待检测图像中电动摩托车的位置。本发明专利技术所采用的可变形部件模型构建了图像特征金字塔，可在不同尺度上提取特征，以适应不同尺度的图像检测；本发明专利技术所采用的可变形部件模型是采用了多组件混合模型，可适应不同视角变化、光照等。

【技术实现步骤摘要】
基于可变部件模型的电动摩托车图像检测方法
本专利技术涉及一种图像检测方法，具体涉及一种基于可变部件模型的电动摩托车图像检测方法，用以在交通视频监控图像中检测出电动摩托车位置。
技术介绍
当前交通视频监控图像趋于复杂多样，存在视角变化、光照变化、背景复杂、目标形态多样化以及噪声干扰等问题，使得在图像中对特定目标进行检测成为一个难点。现有技术中有许多针对行人、车辆、自行车等的目标检测算法，但在电动摩托车的检测识别方面的研究很少，但由电动摩托车所引起的交通事故、偷盗、火灾等事故却不少。目前的车辆识别算法，采用是SIFT特征，通过抽取各个样本中部分特征构成一个特征空间，对其进行稀疏编码获得稀疏编码特征基矩阵，利用线性SVM和稀疏编码结合训练出模板，然后，对目标图像和模板进行SIFT特征匹配得到最优匹配位置，从而检测出车辆。但是该算法存在的缺陷是：(1)主要针对于小汽车、皮卡、大货车、公交车等大型目标，对电动摩托车的识别效果不好。(2)易受外界条件影响：对于遮挡、不同角度拍摄等效果不理想，同时易受光照影响；(3)受目标自身影响：当目标发生形变时，易造成漏检或错检。
技术实现思路
针对当前现有技术中存在的问题，本专利技术的目的是提供一种基于可变部件模型的电动摩托车图像检测方法，将可变部件模型应用于电动摩托车检测，采用多组件模型，可以适应不同视角，提高目标检测准确率。为了实现上述任务，本专利技术采用以下技术方案：基于可变部件模型的电动摩托车图像检测方法，包括以下步骤：步骤1，建立正样本集和负样本集；步骤2，对正样本集分类，负样本集分组，然后对每个正样本、负样本进行特征提取，得到正样本的特征以及负样本的特征；训练所述正样本的特征和负样本的特征，得到根模板；步骤3，将所述的根模板通过空间聚合得到初始根模板，对所有的正样本、负样本分别建立特征金字塔，然后训练所述的初始根模板；步骤4，设定电动摩托车的部件个数和形状，利用部件扫描步骤3训练后的初始根模板，当部件形状与初始根模板覆盖区域的能量最大时，该覆盖区域即为部件的位置，从而得到初始部件模型；将初始部件模型加载到训练后的初始跟模板中，并在所述的正样本集和负样本集上进行训练，得到最终根模板以及最终部件模型；步骤5，对于待检测图像，对待检测图像进行特征提取，然后与所述的最终根模板以及最终部件模型进行匹配，得到待检测图像中电动摩托车的位置。进一步地，所述的建立正样本集和负样本集，包括：将多张包含电动摩托车的图像作为正样本集，将正样本集中每一个正样本中的电动摩托车进行矩形框标注；负样本集则选取PASCALVOC数据集提供的不包含电动摩托车的图像。进一步地，所述的步骤2包括：步骤2.1，将正样本集中的正样本的矩形标注框按照长宽比分为m类，负样本集随机分成m组；步骤2.2，对每个正样本、负样本进行特征提取，包括：对于正样本或负样本图像，将图像分为多个单元格，对单元格内的每个像素分18个有符号梯度方向和9个无符号梯度方向求梯度，然后用梯度方向在单元格的直方图中加权投影，得到该单元格的梯度方向直方图，即为该单元格对应的27维的特征向量；然后将每2*2个相邻单元格组成一个区块，则每个区块对应于一个108维的特征向量，通过对其进行主成分分析和解析降维，将原来的108维特征分别对每行每列进行累加降至4+27＝31维，将所有区块特征向量串联起来即可得到整个图像的特征；由此得到正样本的特征和负样本的特征；步骤2.3，根据正样本的特征和负样本的特征，用SVM训练得到m个根模板F1,F2,…,Fm。进一步地，所述的对所有的正样本、负样本分别建立特征金字塔，包括：对于正样本或负样本图像，将图像作为特征金字塔的第一层，然后特征金字塔的其他层均以上一层21/interval的倍数进行采样，对于特征金字塔的每一层，按照步骤2.2的方法进行特征提取，从而得到特征金字塔每一层的特征。进一步地，进行目标检测，所述的对待检测图像进行特征提取，然后与所述的最终根模板以及最终部件模型进行匹配，得到待检测图像中电动摩托车的位置，包括：建立待检测图像的特征金字塔，将该特征金字塔的每一层与所述最终根模板分别进行卷积操作，选择最大的卷积值作为根模板得分；从待检测图像的特征金字塔中，选择待检测图像的两倍分辨率特征与每一个最终部件模型分别进行卷积，得到每一个最终部件模型对应的得分；计算最终部件模型的变形花费；求所述根模板得分、最终部件模型对应的得分与变形花费的总的代数和作为检测总得分，得分最高的位置即为电动摩托车的位置。本专利技术与现有技术相比具有以下技术特点：1.本专利技术所采用的可变形部件模型构建了图像特征金字塔，可在不同尺度上提取特征，以适应不同尺度的图像检测；2.本专利技术所采用的可变形部件模型是采用了多组件混合模型，可适应不同视角变化、光照等；3.本专利技术所采用的可变形部件模型采用了两倍分辨率的部件模型，可解决小范围遮挡问题。附图说明图1为本专利技术方法的流程图；图2的(a)为最终根模板的示意图，(b)为最终部件模型的示意图，(c)为变形花费的示意图。图3为利用本专利技术方法进行电动摩托车检测的结果示意图。具体实施方式本专利技术公开了一种基于可变部件模型的电动摩托车图像检测方法，包括以下步骤：步骤1，建立正样本集和负样本集将多张包含电动摩托车的图像作为正样本集P，将正样本集中每一个正样本中的电动摩托车进行矩形框标注，本方案采用的是弱标注的方式，即仅对目标位置(即电动摩托车所在位置)进行标注；本实施例中，正样本集采用1000张拍摄的包含电动摩托车的的图像；负样本集N则选取PASCALVOC数据集提供的1000张不包含电动摩托车的图像。步骤2，对正样本集分类，负样本集分组，然后对每个正样本、负样本进行特征提取，得到正样本的特征以及负样本的特征；训练所述正样本的特征和负样本的特征，得到根模板；具体步骤如下：步骤2.1，将正样本集中的正样本的矩形标注框按照长宽比分为m类，负样本集随机分成m组；步骤2.2，对每个正样本、负样本进行特征提取，包括：对于正样本或负样本图像(该图像即指正样本、负样本)，将图像分为多个单元格，本实施例中每个单元格为8*8像素大小；对单元格内的每个像素分18个有符号梯度方向(0°-360°)和9个无符号梯度方向(0°-180°)求梯度，然后用梯度方向在单元格的直方图中加权投影，得到该单元格的梯度方向直方图，即为该单元格对应的18+9＝27维的特征向量；然后将每2*2个相邻单元格(上下左右四个单元格)组成一个区块(如上下左右相邻不足四个单元格，则在缺失的部分补0)，则每个区块对应于一个4*27＝108维的特征向量，通过对其进行主成分分析和解析降维，将原来的108维特征分别对每行每列进行累加降至4+27＝31维，将所有区块特征向量串联起来即可得到整个图像的特征。对每一个正样本的图像、负样本的图像都按照上述方法进行处理，即得到了正样本的特征和负样本的特征。步骤2.3，根据所有的正样本的特征和负样本的特征，用标准支持向量机SVM训练得到m个根模板F1,F2,…,Fm。步骤3，将所述的根模板通过空间聚合得到初始根模板，对所有的正样本、负样本分别建立特征金字塔，然后训练所述的初始根模板；具体包括：步骤3.1，将所述的m个根模板F1,F2,…,Fm通本文档来自技高网...

【技术保护点】
1.基于可变部件模型的电动摩托车图像检测方法，其特征在于，包括以下步骤：步骤1，建立正样本集和负样本集；步骤2，对正样本集分类，负样本集分组，然后对每个正样本、负样本进行特征提取，得到正样本的特征以及负样本的特征；训练所述正样本的特征和负样本的特征，得到根模板；步骤3，将所述的根模板通过空间聚合得到初始根模板，对所有的正样本、负样本分别建立特征金字塔，然后训练所述的初始根模板；步骤4，设定电动摩托车的部件个数和形状，利用部件扫描步骤3训练后的初始根模板，当部件形状与初始根模板覆盖区域的能量最大时，该覆盖区域即为部件的位置，从而得到初始部件模型；将初始部件模型加载到训练后的初始跟模板中，并在所述的正样本集和负样本集上进行训练，得到最终根模板以及最终部件模型；步骤5，对于待检测图像，对待检测图像进行特征提取，然后与所述的最终根模板以及最终部件模型进行匹配，得到待检测图像中电动摩托车的位置。

【技术特征摘要】
1.基于可变部件模型的电动摩托车图像检测方法，其特征在于，包括以下步骤：步骤1，建立正样本集和负样本集；步骤2，对正样本集分类，负样本集分组，然后对每个正样本、负样本进行特征提取，得到正样本的特征以及负样本的特征；训练所述正样本的特征和负样本的特征，得到根模板；步骤3，将所述的根模板通过空间聚合得到初始根模板，对所有的正样本、负样本分别建立特征金字塔，然后训练所述的初始根模板；步骤4，设定电动摩托车的部件个数和形状，利用部件扫描步骤3训练后的初始根模板，当部件形状与初始根模板覆盖区域的能量最大时，该覆盖区域即为部件的位置，从而得到初始部件模型；将初始部件模型加载到训练后的初始跟模板中，并在所述的正样本集和负样本集上进行训练，得到最终根模板以及最终部件模型；步骤5，对于待检测图像，对待检测图像进行特征提取，然后与所述的最终根模板以及最终部件模型进行匹配，得到待检测图像中电动摩托车的位置。2.如权利要求1所述的.基于可变部件模型的电动摩托车图像检测方法，其特征在于，所述的建立正样本集和负样本集，包括：将多张包含电动摩托车的图像作为正样本集，将正样本集中每一个正样本中的电动摩托车进行矩形框标注；负样本集则选取PASCALVOC数据集提供的不包含电动摩托车的图像。3.如权利要求1所述的.基于可变部件模型的电动摩托车图像检测方法，其特征在于，所述的步骤2包括：步骤2.1，将正样本集中的正样本的矩形标注框按照长宽比分为m类，负样本集随机分成m组；步骤2.2，对每个正样本、负样本进行特征提取，包括：对于正样本或负样本图像，将图像分为多个单元格，对单元格内的每个像素分18个有符号梯度方向和9个无符号...

【专利技术属性】
技术研发人员：刘玉萍，孙盛，陈平华，余旭，董晓冬，
申请(专利权)人：广东工业大学，
类型：发明
国别省市：广东,44