一种彩绘列车车号识别方法、系统及介质转让专利

申请号 : CN202210744590.3

文献号 : CN114821452B

文献日 :

基本信息:

PDF:

法律信息:

相似专利:

发明人 : 庞先昂孙振行乔文静董利亚

申请人 : 山东博昂信息科技有限公司

摘要 :

本发明涉及一种彩绘列车车号识别方法、系统及介质,属于图像处理技术领域;彩绘列车车号识别方法包括S1、获取监控采集的列车视频数据流,通过视频数据流获取帧间信息,根据帧间信息获取列车运动变化区域的图像;S2、对图像进行预处理,通过高斯滤波去除噪音干扰;S3、对去除噪音干扰后的图像进行车号字符区域定位,得到包含车号字符区域的分割图像;S4、对包含车号字符区域的分割图像进行HSL变换处理;S5、通过轻量化CRNN网络模型对进行了HSL变换处理后的图像进行字符识别;本发明对于彩绘的列车的车号实现自动识别,提高了列车车号识别的精确度,减少车号识别的时间,自动化程度高。

权利要求 :

1.一种彩绘列车车号识别方法,其特征在于,所述方法包括以下步骤:获取监控采集的列车视频数据流,通过所述视频数据流获取帧间信息,根据所述帧间信息获取列车运动变化区域的图像;

对图像进行预处理,去除噪音干扰;

对去除噪音干扰后的图像进行车号字符区域定位,得到包含车号字符区域的分割图像;

对分割图像进行HSL变换,得到分割图像的亮度值L,根据得到的亮度值L,去掉分割图像中与黑色车号不同的颜色,留下黑色车号字符信息,具体分割图像为RGB图片,设RGB的颜色为(LR,LG,LB),设Lmax为LR,LG,LB的最大值,Lmin为LR,LG,LB的最小值,LR,LG,LB的取值区间为[0,1];

(1)亮度L的计算公式:L=(Lmax+Lmin)/2,特殊情形,当Lmax=Lmin,此时有Lmax=LR=LG=LB=Lmin,标明颜色为灰色,此时S=0,H不表示任何颜色;

(2)饱和度S的计算,分两种情况:

若L≤0.5,则S=(Lmax‑Lmin)/(Lmax+Lmin);

若L>0.5,则S=(Lmax‑Lmin)/(2‑Lmax‑Lmin);

(3)色调H的计算公式如下,分三种情形:

当Lmax=LR时,H=60*(LG‑LB)/(Lmax+Lmin),颜色处于黄色和品红之间;

当Lmax=LG时,H=120+60*(LB‑LR)/(Lmax+Lmin),颜色处于青色和黄色之间;

当Lmax=LB时,H=240+60*(LR‑LG)/(Lmax+Lmin),颜色处于品红和青色之间;

如果上述计算H出现负值,则增加360,因为H为周期函数;HSL变换后,统计每个像素点的亮度值,亮度值越高,色彩越白,亮度值越低,色彩越黑,选择合适的阈值,作为临界点,建立图像模版,亮度值大于阈值的点,图像模版中像素点的值为255,否则为0,然后,用图像模版与原RGB图做匹配,这样只保存黑色车号信息;

将进行了HSL变换处理后的图像发送至轻量化的CRNN网络模型,进行字符识别。

2.根据权利要求1所述的一种彩绘列车车号识别方法,其特征在于:所述对图像进行预处理,去除噪音干扰具体如下:采用滤波算法,通过高斯滤波去除噪音干扰。

3.根据权利要求2所述的一种彩绘列车车号识别方法,其特征在于,所述对去除噪音干扰后的图像进行车号字符区域定位,得到包含车号字符区域的分割图像,步骤具体包括:获取去除噪音干扰后的图像,并将其发送至预设的字符区域定位模型;

利用所述预设的字符区域定位模型对去除噪音干扰后的图像进行车号字符区域定位;

将定位出车号字符区域的图像进行分割,得到包含车号字符区域的分割图像。

4.根据权利要求3所述的一种彩绘列车车号识别方法,其特征在于,所述字符区域定位模型的具体生成步骤如下:获取初始模型和第一数据集,所述第一数据集包括常规列车图片和彩绘列车图片;

对所述列车的车号图片中的字符进行区域标注,构成第一训练集和第一测试集;

根据第一训练集和第一测试集对初始模型进行目标检测网络训练,生成字符区域定位模型。

5.根据权利要求4所述的一种彩绘列车车号识别方法,其特征在于,所述轻量化的CRNN网络模型的具体生成步骤如下:获取第二数据集,所述第二数据集为常规列车图片和彩绘列车图片;

通过字符区域定位模型对第二数据集进行处理得到车号字符区域定位分割图片,以及利用分割图片HSL变换信息对分割图片进行处理后的图片,构成第二训练集和第二测试集;

根据第二训练集和第二测试集进行网络训练,生成轻量化的CRNN网络模型。

6.一种彩绘列车车号识别系统,其特征在于,包括:视频采集模块(1),用于获取监控采集的列车视频数据流,通过所述视频数流获取帧间信息,根据所述帧间信息获取列车运动变化区域的图像;

噪音去除模块(2),用于对图像进行预处理,去除噪音干扰;

字符定位模块(3),用于对去除噪音干扰后的图像进行车号字符区域定位,得到包含车号字符区域的分割图像;

HSL变换模块(4),用于对分割图像进行HSL变换,得到分割图像的亮度值L,根据得到的亮度值L,去掉分割图像中与黑色车号不同的颜色,留下黑色车号字符信息,具体分割图像为RGB图片,设RGB的颜色为(LR,LG,LB),设Lmax为LR,LG,LB的最大值,Lmin为LR,LG,LB的最小值,LR,LG,LB的取值区间为[0,1];

(1)亮度L的计算公式:L=(Lmax+Lmin)/2,特殊情形,当Lmax=Lmin,此时有Lmax=LR=LG=LB=Lmin,标明颜色为灰色,此时S=0,H不表示任何颜色;

(2)饱和度S的计算,分两种情况:

若L≤0.5,则S=(Lmax‑Lmin)/(Lmax+Lmin);

若L>0.5,则S=(Lmax‑Lmin)/(2‑Lmax‑Lmin);

(3)色调H的计算公式如下,分三种情形:

当Lmax=LR时,H=60*(LG‑LB)/(Lmax+Lmin),颜色处于黄色和品红之间;

当Lmax=LG时,H=120+60*(LB‑LR)/(Lmax+Lmin),颜色处于青色和黄色之间;

当Lmax=LB时,H=240+60*(LR‑LG)/(Lmax+Lmin),颜色处于品红和青色之间;

如果上述计算H出现负值,则增加360,因为H为周期函数;HSL变换后,统计每个像素点的亮度值,亮度值越高,色彩越白,亮度值越低,色彩越黑,选择合适的阈值,作为临界点,建立图像模版,亮度值大于阈值的点,图像模版中像素点的值为255,否则为0,然后,用图像模版与原RGB图做匹配,这样只保存黑色车号信息;

字符识别模块(5),用于将进行了HSL变换处理后的图像发送至轻量化的CRNN网络模型,进行字符识别。

7.一种存有计算机程序的计算机可读存储介质,其特征在于,存储有实现权利要求1‑5任一方法的计算机程序。

说明书 :

一种彩绘列车车号识别方法、系统及介质

技术领域

[0001] 本发明涉及图像处理技术领域,尤其是涉及一种彩绘列车车号识别方法、系统及介质。

背景技术

[0002] 目前城镇化越来越深入,列车普及度增加,列车的安全问题随之加强,列车停站时会针对列车进行检修,对列车进行检修时,需要获取列车的车号。
[0003] 现在随着技术的发展,促进地方文化的宣传,列车车体上出现了彩绘图案。例如白鹈鹕、水蕨菜、红木棉、蜂虎鸟等。这些彩绘图案,对车体上的部分车号进行了遮挡,有的彩绘图案的颜色与车号的颜色相近,这大大增加了车号识别的难度。

发明内容

[0004] 为了解决上述背景技术中提出的问题,本发明提供一种彩绘列车车号识别方法、系统及介质。
[0005] 第一方面,本发明申请提供一种彩绘列车车号识别方法,采用如下的技术方案:
[0006] 一种彩绘列车车号识别方法,包括以下步骤:
[0007] 获取监控采集的列车视频数据流,通过所述视频数据流获取帧间信息,根据所述帧间信息获取列车运动变化区域的图像;
[0008] 对图像进行预处理,去除噪音干扰;
[0009] 对去除噪音干扰后的图像进行车号字符区域定位,得到包含车号字符区域的分割图像;
[0010] 对分割图像进行HSL变换处理;
[0011] 对进行了HSL变换处理后的图像进行字符识别。
[0012] 通过采用上述技术方案,通过获取监控采集的列车视频数据流,首先根据帧间信息获取列车运动变化的区域,得到列车的图像,对图像进行预处理去除噪音干扰,提高图像的清晰度和质量,对去除噪音干扰后的图像进行车号字符区域定位,并进行截取,得到包含车号字符区域的分割图像,对分割图像进行HSL变换,可以得到图像的亮度,亮度值越高,色彩越白,亮度值越低,色彩越黑,根据亮度值,去掉分割图像中与黑色车号不同的颜色,留下黑色车号字符图像,进行识别,自动化程度高,降低了车号识别的难度,可以有效识别列车的车号,提高了列车车号识别的精确度,且减少了列车车号识别的时间,提高了列车检修的效率。
[0013] 优选的,所述对图像进行预处理,去除噪音干扰具体如下:
[0014] 采用滤波算法,通过高斯滤波去除噪音干扰。通过采用滤波算法可以有效去除噪音干扰,提高图像的清晰度和质量,便于目标定位、识别等后续工作的进行。
[0015] 优选的,所述对去除噪音干扰后的图像进行车号字符区域定位,得到包含车号字符区域的分割图像,步骤具体包括:
[0016] 获取去除噪音干扰后的图像,并将其发送至预设的字符区域定位模型;
[0017] 利用所述预设的字符区域定位模型对去除噪音干扰后的图像进行车号字符区域定位;
[0018] 将定位出车号字符区域的图像进行分割,得到包含车号字符区域的分割图像。
[0019] 优选的,所述字符区域定位模型的具体生成步骤如下:
[0020] 获取初始模型和第一数据集,所述第一数据集包括常规列车图片和彩绘列车图片;
[0021] 对所述列车的车号图片中的字符进行区域标注,构成第一训练集和第一测试集;
[0022] 根据第一训练集和第一测试集对初始模型进行目标检测网络训练,生成字符区域定位模型。
[0023] 通过将定位出车号字符区域的图像分割出来,减少非车号字符区域对字符识别的影响,提高车号识别的精准度。
[0024] 优选的,所述对分割图像进行HSL变换处理,步骤具体包括:
[0025] 根据包含有车号字符区域的分割图像,将所述分割图像进行HSL变换,根据HSL变换中的L信息,对分割图像进行处理。
[0026] 通过将图像进行HSL变换,可以得到图像的亮度L,即色彩的明暗程度,亮度值越高,色彩越白,亮度值越低,色彩越黑,根据得到的亮度值可以去掉分割图像中与黑色车号不同的颜色,留下黑色车号字符图像。
[0027] 优选的,所述对进行了HSL变换处理后的图像进行字符识别,步骤具体包括:
[0028] 获取进行HSL变换处理后的图像,并将其发送至预先生成的轻量化的CRNN网络模型;
[0029] 利用所述预先生成的轻量化的CRNN网络模型,对进行HSL变换处理后的图像中的字符进行识别。
[0030] 通过采用轻量化CRNN网络模型,模型量化后模型变小,识别时间缩短。
[0031] 优选的,所述轻量化CRNN网络模型的具体生成步骤如下:
[0032] 获取第二数据集,所述第二数据集为常规列车图片和彩绘列车图片;
[0033] 通过字符区域定位模型对第二数据集进行处理,得到包含车号字符区域的分割图像,以及对分割图像进行HSL变换,利用亮度L信息对分割图像进行处理后得到的图像,构成第二训练集和第二测试集;
[0034] 根据第二训练集和第二测试集进行网络训练,生成轻量化的CRNN网络模型。
[0035] 第二方面,本发明申请提供一种彩绘列车车号识别的系统,采用如下的技术方案:
[0036] 一种彩绘列车车号识别的系统,包括:
[0037] 视频采集模块,用于获取列车行驶的视频数据,通过所述视频数据获取帧间信息,根据所述帧间信息获取列车运动变化区域的图像;
[0038] 噪音去除模块,用于将获取的列车运动变化区域的图像进行预处理,去除噪音干扰;
[0039] 字符定位模块,将去除噪音干扰后的图像进行车号字符区域定位,得到包含车号字符区域的分割图像;
[0040] HSL变换模块,基于包含车号字符区域的分割图像,将其进行HSL变换处理;
[0041] 字符识别模块,将进行HSL变换处理后的图像进行字符识别。
[0042] 通过采用上述技术方案,通过视频采集模块获取列车的视频数据,首先根据帧间信息获取列车运动变化的区域,得到列车的图像,通过噪音去除模块对图像进行预处理去除噪音干扰,提高图像的清晰度和质量,通过字符定位模块对去除噪音干扰后的图像进行车号字符区域定位,并进行截取,得到包含车号字符区域的分割图像,通过HSL变换模块将分割图像进行HSL变换,可以得到图像的亮度,亮度值越高,色彩越白,亮度值越低,色彩越黑,根据亮度值,去掉分割图像中与黑色车号不同的颜色,留下黑色车号字符图像,并发送至字符识别模块进行识别,自动化程度高,降低了车号识别的难度,可以有效识别列车的车号,提高了列车车号识别的精确度,且减少了列车车号识别的时间,提高了列车检修的效率。
[0043] 第三方面,本发明申请提供一种存有计算机程序的计算机可读存储介质,采用如下的技术方案:
[0044] 一种存有计算机程序的计算机可读存储介质,存储有实现第一方面中任一方法的计算机程序。
[0045] 通过采用上述技术方案,通过获取监控采集的列车视频数据流,首先根据帧间信息获取列车运动变化的区域,得到列车的图像,对图像进行预处理去除噪音干扰,提高图像的清晰度和质量,对去除噪音干扰后的图像进行车号字符区域定位,并进行截取,得到包含车号字符区域的分割图像,对分割图像进行HSL变换,可以得到图像的亮度,亮度值越高,色彩越白,亮度值越低,色彩越黑,根据亮度值,去掉分割图像中与黑色车号不同的颜色,留下黑色车号字符图像,将其进行识别,自动化程度高,降低了车号识别的难度,可以有效识别列车的车号,提高了列车车号识别的精确度,且减少了列车车号识别的时间,提高了列车检修的效率。
[0046] 综上所述,本发明具有如下的有益技术效果:
[0047] 1.本发明通过获取监控采集的列车视频数据流,首先根据帧间信息获取列车运动变化的区域,得到列车的图像,对图像进行预处理去除噪音干扰,通过预设的字符区域定位模型对去除噪音干扰后的图像进行车号字符区域定位并进行截取,得到包含有车号字符区域的分割图像,把分割图像进行HSL变换,利用L亮度信息对分割图像进行处理,得到黑色车号字符图像,通过预设的轻量化的CRNN网络模型进行识别,可以准确识别车号;
[0048] 2.本发明自动化程度高,降低了车号识别的难度,可以有效识别列车的车号,提高列车车号识别的精确度,且减少了列车车号识别的时间,提高了列车检修的效率。

附图说明

[0049] 图1是本发明其中一实施例示出的彩绘列车车号识别方法的流程图;
[0050] 图2是本发明其中一实施例示出的彩绘列车车号识别方法的结构框图;
[0051] 图3为轻量化CRNN网络模型图。
[0052] 附图标记说明:
[0053] 1、视频采集模块;
[0054] 2、噪音去除模块;
[0055] 3、字符定位模块;
[0056] 4、HSL变换模块;
[0057] 5、字符识别模块。

具体实施方式

[0058] 以下结合图1‑图3对本发明作进一步详细说明。
[0059] 本发明实施例公开一种彩绘列车车号识别方法。
[0060] 在一可选的实施方式中,识别方法如图1所示,具体步骤如下:
[0061] S1、获取监控采集的列车视频数据流,通过视频数据流获取帧间信息,根据帧间信息获取列车运动变化区域的图像;
[0062] 利用前后帧间信息的相关性获取动态图像,便于进行图像字符区域定位并识别。
[0063] S2、对图像进行预处理,去除噪音干扰;
[0064] 采用滤波算法去除噪音干扰,具体为采用高斯滤波去除噪音干扰,高斯滤波就是对整个图像进行加权平均的过程,每一个像素点的值,都由其本身和邻域内的其他像素值经过加权平均后得到。
[0065] 对图像去除噪音干扰后,可以提高图像的清晰度和质量,便于图像字符区域定位、识别等后续工作的进行。
[0066] S3、对去除噪音干扰后的图像进行车号字符区域定位,得到包含车号字符区域的分割图像;
[0067] 通过将定位出车号字符区域的图像分割出来,减少非车号字符区域对字符识别的影响,提高车号识别的精准度。
[0068] 其中,步骤S3具体包括:
[0069] 获取去除噪音干扰后的图像,并将其发送至预设的字符区域定位模型;
[0070] 利用所述预设的字符区域定位模型对去除噪音干扰后的图像进行车号字符区域定位;
[0071] 将定位出车号字符区域的图像进行分割,得到包含车号字符区域的分割图像。
[0072] 其中字符区域定位模型的具体生成步骤如下:
[0073] 获取初始模型和第一数据集,第一数据集包括常规列车图片和彩绘列车图片;
[0074] 对列车的车号图片中的字符进行区域标注,构成第一训练集和第一测试集;
[0075] 根据第一训练集和第一测试集对初始模型进行目标检测网络训练,生成字符区域定位模型;
[0076] 其中对列车的车号图片中的字符进行区域标注,区域是指字符在图像中的位置,区域标注就是对字符所在的位置通过区域标注软件进行标注,区域标注软件可以选用labelme。
[0077] 需要说明的是,目标检测网络的训练过程分为两个阶段,第一个阶段是数据由低层次向高层次传播的阶段,即前向传播阶段。另外一个阶段是,当前向传播得出的结果与预期不相符时,将误差从高层次向低层次进行传播训练的阶段,即反向传播阶段。具体过程如下:
[0078] 1、网络进行权值的初始化;
[0079] 2、输入数据经过网络向前传播得到输出值;
[0080] 3、求出网络的输出值与目标值之间的误差;
[0081] 4、当误差大于我们的期望值时,将误差传回网络中,反向传播更新权值。
[0082] 5、重复进行2‑4步,直到误差等于或小于我们的期望值时,结束训练。
[0083] S4、对分割图像进行HSL变换处理;
[0084] 根据包含有车号字符区域的分割图像,分割图像为RGB图片,将分割图像进行HSL变换,根据HSL变换中的L信息,对分割图像进行处理。
[0085] 通过将图像进行HSL变换,可以得到图像的亮度L,即色彩的明暗程度,亮度值越高,色彩越白,亮度值越低,色彩越黑,根据得到的亮度值可以去掉分割图像中与黑色车号不同的颜色,留下黑色车号字符信息。
[0086] 具体处理过程如下:
[0087] 设RGB的颜色为(LR,LG,LB),设Lmax为LR,LG,LB的最大值,Lmin为LR,LG,LB的最小值,LR,LG,LB的取值区间为[0,1];
[0088] (1)亮度L的计算公式:L=(Lmax+Lmin)/2
[0089] 特殊情形,当Lmax=Lmin,此时有Lmax=LR=LG=LB=Lmin,标明颜色为灰色,此时S=0,H不表示任何颜色;
[0090] (2)饱和度S的计算,分两种情况:
[0091] 若L≤0.5,则S=(Lmax‑Lmin)/(Lmax+Lmin);
[0092] 若L>0.5,则S=(Lmax‑Lmin)/(2‑Lmax‑Lmin);
[0093] (3)色调H的计算公式如下,分三种情形:
[0094] 当Lmax=LR时,H=60*(LG‑LB)/(Lmax+Lmin),颜色处于黄色和品红之间;
[0095] 当Lmax=LG时,H=120+60*(LB‑LR)/(Lmax+Lmin),颜色处于青色和黄色之间;
[0096] 当Lmax=LB时,H=240+60*(LR‑LG)/(Lmax+Lmin),颜色处于品红和青色之间;
[0097] 如果上述计算H出现负值,则增加360,因为H为周期函数。
[0098] 其中,HSL即色相、饱和度、亮度(英语:Hue,Saturation,Lightness),[0099] 色相(H)是色彩的基本属性,就是平常所说的颜色名称,如红色、黄色等;饱和度(S)是指色彩的纯度,越高色彩越纯,低则逐渐变灰,取0‑100%的数值;亮度(L)指的是色彩的明暗程度,亮度值越高,色彩越白,亮度越低,色彩越黑;明度(V)和亮度(L),取0‑100%;
[0100] HSL变换后,统计每个像素点的亮度值,亮度值越高,色彩越白,亮度值越低,色彩越黑,选择合适的阈值,作为临界点,建立图像模版(亮度值大于阈值的点,图像模版中像素点的值为255,否则为0),然后,用图像模版与原RGB图做匹配,这样只保存黑色车号信息。
[0101] S5、对进行了HSL变换处理后的图像进行字符识别;
[0102] 步骤S5具体包括:
[0103] 获取进行HSL变换处理后的图像,并将其发送至预先生成的轻量化的CRNN网络模型;
[0104] 利用预先生成的轻量化的CRNN网络模型,对进行HSL变换处理后的图像中的字符进行识别;
[0105] 其中,轻量化的CRNN网络模型的具体生成步骤如下:
[0106] 获取第二数据集,第二数据集为常规列车图片和彩绘列车图片;
[0107] 通过字符区域定位模型对第二数据集进行处理,得到包含车号字符区域的分割图像,以及对分割图像HSL变换,利用亮度L信息对分割图像进行处理后得到的图像,构成第二训练集和第二测试集;
[0108] 根据第二训练集和第二测试集进行网络训练,生成轻量化的CRNN网络模型;
[0109] 对字符识别模型进行轻量化,得到轻量化CRNN网络模型,轻量化CRNN网络模型包含三部分,从下到上依次为:
[0110] (1)卷积层:从输入图像中提取出特征序列;
[0111] (2)循环层:预测从卷积层获取的特征序列的标签分布;
[0112] (3)转录层:把从循环层获取的标签分布通过去重、整合等操作转换成最终的识别结果。
[0113] 如图3所示为轻量化CRNN网络模型图。
[0114] 轻量化CRNN网络模型,轻量化后模型小,字符识别时间缩短。
[0115] 基于上述彩绘列车车号识别方法,本申请实施例还公开了一种彩绘列车车号识别系统。
[0116] 在一可选的实施方式中,具体包括:
[0117] 视频采集模块1,用于获取列车行驶的视频数据,通过视频数据获取帧间信息,根据帧间信息获取列车运动变化区域的图像;
[0118] 噪音去除模块2,用于将获取的列车运动变化区域的图像进行预处理,去除噪音干扰;
[0119] 字符定位模块3,将去除噪音干扰后的图像进行车号字符区域定位,得到包含车号字符区域的分割图像;
[0120] HSL变换模块4,基于包含车号字符区域的分割图像,将其进行HSL变换处理;
[0121] 字符识别模块5,将进行HSL变换处理后的图像进行字符识别。
[0122] 本发明实施例的工作过程通过视频采集模块1获取列车行驶的视频数据,从视频数据获取帧间信息,根据帧间信息获取列车运动变化区域的图像;将列车运动变化区域的图像发送至噪音去除模块2对其去除噪音干扰,再将去除噪音干扰后的图像发送至字符定位模块3对其进行车号字符区域定位,根据定位出的车号字符区域位置进行截取,得到包含车号字符区域的分割图像,将分割图像发送至HSL变换模块4,进行HSL变换,根据HSL变换中的L信息,对分割图像进行处理,去掉分割图像中与车号不同的颜色,留下车号字符图像,并将其发送至字符识别模块5识别出车号字符。
[0123] 本申请实施例还公开了一种存有计算机程序的计算机可读存储介质,存储有实现上述任一方法的计算机程序。
[0124] 以上均为本发明的较佳实施例,并非依此限制本发明的保护范围,故:凡依本发明的结构、形状、原理所做的等效变化,均应涵盖于本发明的保护范围之内。