一种自动训练形状匹配模型的方法技术

技术编号:30013522 阅读:27 留言:0更新日期:2021-09-11 06:16
本发明专利技术提出了一种自动训练形状匹配模型的方法,包括:步骤S1,采集一张没有目标的图像;步骤S2,采集一张包含目标的图像;步骤S3,获取步骤S1和步骤S2采集的两张图像的多个差异区域;步骤S4,计算每个区域的最小外接矩形或椭圆形,以及其长轴与图像x轴形成的角度;步骤S5,针对所有差异区域提取边缘点,并计算每个边缘点基于灰度的梯度信息;步骤S6,随机选取一个差异区域中的边缘点训练初始模型;步骤S7,通过其他区域中的边缘点对上述模型进行修饰,得到适用于所有目标的模型;步骤S8,利用修改后的模型在步骤S2采集的图像中搜索目标,如果无法找到所有目标或到达最大迭代次数,则返回步骤S5,最后一次迭代得到的模型即为最终模型。型。型。

【技术实现步骤摘要】
一种自动训练形状匹配模型的方法


[0001]本专利技术涉及图像定位处理
,特别涉及一种自动训练形状匹配模型的方法。

技术介绍

[0002]模板匹配是一种在图像中搜索另一幅图像(称之为模板图像)的技术,广泛用于在图像中定位目标。其基本思路为:首先根据模板图像的特征训练模型,然后在新的图像中利用模型信息搜索模板图像,最终得到目标在新的图像中的位置。
[0003]在当前机器视觉软件中,模板匹配是常用的功能,其第一步是获取模板图像然后训练模型。训练模型的流程为:1.将需要定位的目标放置于视野中并获取包含目标的图像;2.在图像中设置包含目标的ROI并将ROI中的部分作为模板图像;3.设置模型训练参数。上述过程要求人去参与ROI和参数的设置,相对繁琐且需要人了解参数的意义,无法实现模型训练的完全自动化。
[0004]模板匹配根据匹配原理的不同有多种匹配方法,《OCCLUSION,CLUTTER,AND ILLUMINATION INVARIANT OBJECT RECOGNITION》提出了一种工业级的模板匹配方法——基于形状的匹配(简称形状匹配),在非线性光源变化以及存在遮挡、污损等情况下依然可以实现很好的识别。其基本原理为提取模板图像中的边缘点,并计算每个边缘点的梯度,利用所有边缘点所包含的位置和梯度信息以及目标可能发生的变化(平移、旋转和缩放)信息计算模型,随后在其他图像中搜索该模型。可见,该方法所需要设置的参数也比较多,需要一定的专业知识,具有较高的使用门槛。

技术实现思路

[0005]本专利技术的目的旨在至少解决所述技术缺陷之一。
[0006]为此,本专利技术的目的在于提出一种自动训练形状匹配模型的方法。
[0007]为了实现上述目的,本专利技术的实施例提供一种自动训练形状匹配模型的方法,包括如下步骤:
[0008]步骤S1,采集一张没有目标的图像;
[0009]步骤S2,采集一张包含目标的图像,其中,所述步骤S1采集的图像和步骤S2采集的图像的区别仅在于是否存在目标;
[0010]步骤S3,获取步骤S1和步骤S2采集的两张图像的多个差异区域;
[0011]步骤S4,计算每个区域的最小外接矩形或椭圆形;
[0012]步骤S5,针对所有差异区域提取边缘点,并计算每个边缘点基于灰度的梯度信息;
[0013]步骤S6,随机选取一个差异区域中的边缘点训练初始模型;
[0014]步骤S7,通过其他区域中的边缘点对上述模型进行修饰,得到适用于所有目标的模型;
[0015]步骤S8,利用修改后的模型在步骤S2采集的图像中搜索目标,如果无法找到所有
目标或到达最大迭代次数,则返回步骤S5,最后一次迭代得到的模型即为最终模型。
[0016]进一步,所述步骤S3包括:
[0017]步骤S31,对两张图像逐像素作差并取绝对值,得到新的图像;
[0018]步骤S32,对所述新的图像进行二值化,得到二值化图像;
[0019]步骤S33,根据所述二值化图像中像素为255的点之间的连通关系,获得所述二值化图像中的多个连通域;所述多个连通域对应表示多个差异区域。
[0020]进一步,在所述步骤S31中,采用下式计算新的图像:
[0021]Image3(i,j)=|Image1(i,j)

Image2(i,j)|
[0022]采用下式计算二值化图像:
[0023]Image4(i,j)=|Image3(i,j)|>threshold?255:0,其中threshold为参数。
[0024]进一步,计算边缘点,并在边缘提取的过程中同时计算每个边缘点基于灰度的梯度信息。
[0025]进一步,在所述步骤S6中,所述训练初始模型,包括如下步骤:
[0026]步骤S61,确定该区域边缘点的中心点;
[0027]步骤S62,计算该区域边缘点相对该中心点的位置;
[0028]步骤S63,将该区域边缘点绕该中心点以固定步长进行旋转,计算每次旋转后所有新的边缘点位置和梯度,新的边缘点信息构成一个新的集合;
[0029]步骤S64,所有旋转后的集合信息构成初始模型。
[0030]进一步,在所述步骤S61中,使用该区域最小外接矩形或椭圆形的中心作为边缘点的中心点。
[0031]进一步,所述步骤S7,包括:
[0032]步骤S71,从其他区域中选择另一个区域,计算该区域中心点;
[0033]步骤S72,根据当前区域的最小外接矩形或椭圆形的角度选择初始模型中与该角度匹配的模型边缘点集合,该角度成为初始角度θ;
[0034]步骤S73,计算匹配分数s_0;
[0035]步骤S74,选择初始模型中与上述初始角度大一个角度步长的模型边缘点集合;
[0036]步骤S75,重复步骤S73,计算新的匹配分数s_1,若s_1>s_0,则重复步骤S74和步骤S75 n次,直至s_n<s_n

1;
[0037]步骤S76,反向选择初始模型中与上述初始角度小一个角度步长的模型边缘点集合;
[0038]步骤S77,重复步骤S73,计算新的匹配分数s_

1,若s_

1>s_0,则重复步骤S76和步骤S77 m次,直至s_

m<s_

m+1;
[0039]步骤S78,在该区域中心点附近k*k的领域重复步骤S72至步骤S77;
[0040]步骤S79,选择所有s0、s_n

1和s_

m+1中最大值对应的角度α以及中心点在k*k区域中的位置偏移l;
[0041]步骤S80,将α和l,重新计算每个点的梯度向量点积,并删除初始模型中梯度向量点积小于一定阈值的点;
[0042]步骤S81,选择剩下区域中的下一个区域,重复步骤S72至步骤S80。
[0043]根据本专利技术实施例的自动训练形状匹配模型的方法,具有以下有益效果:通过只
拍摄两张图像即可实现形状匹配模型的自动化训练。
[0044]1、提升形状匹配的易用性,降低使用门槛。用户只需要获取两张图像即可,一张无目标图像和一张含有单个或多个目标的图像。
[0045]2、实现参数的自动化设置,降低视觉程序配置的工作量。
[0046]本专利技术附加的方面和优点将在下面的描述中部分给出,部分将从下面的描述中变得明显,或通过本专利技术的实践了解到。
附图说明
[0047]本专利技术的上述和/或附加的方面和优点从结合下面附图对实施例的描述中将变得明显和容易理解,其中:
[0048]图1为根据本专利技术实施例的自动训练形状匹配模型的方法的流程图。
具体实施方式
[0049]下面详细描述本专利技术的实施例,本文档来自技高网
...

【技术保护点】

【技术特征摘要】
n次,直至s_n<s_n

1;步骤S76,反向选择初始模型中与上述初始角度小一个角度步长的模型边缘点集合;步骤S77,重复步骤S73,计算新的匹配分数s_

1,若s_

1>s_0,则重复步骤S76和步骤S77 m次,直至s_

m<s_

m+1;步...

【专利技术属性】
技术研发人员:庹华袁顺宁张彪李亚楠韩峰涛耿旭达任赜宇张雷
申请(专利权)人:珞石山东智能科技有限公司
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1