一种面向复杂交通场景的字符识别方法、装置、设备和介质制造方法及图纸

技术编号:34291758 阅读:32 留言:0更新日期:2022-07-27 09:39
本申请涉及人工智能技术领域,提出了一种面向复杂交通场景的字符识别方法、装置、设备和介质,方法包括:通过检测模型对交通场景图像进行目标检测,得到目标框;再通过分类模型获得第一特征图和目标类型,根据目标类型对第一特征图进行切分,得到字符特征图序列,最后通过识别模型对字符特征图序列进行处理,得到字符识别结果。本申请实施例的方案,通过分类模型对检测到的目标框进行特征提取,生成特征图,然后复用该特征图进行目标分类和字符识别,这样能够减小模型的处理数据量;以及,根据目标类型可以判断特征图中的字符行数,进而根据字符行数可以更加简单高效地完成字符分割,提升了计算速度,且节省了计算资源。且节省了计算资源。且节省了计算资源。

【技术实现步骤摘要】
一种面向复杂交通场景的字符识别方法、装置、设备和介质


[0001]本申请涉及人工智能
,特别是涉及一种面向复杂交通场景的字符识别方法、装置、电子设备和计算机可读存储介质。

技术介绍

[0002]随着智慧交通系统的持续深入发展,交通场景已成为光学字符识别(Optical Character Recognition,OCR)技术的一个重要应用场景,在多项人工智能任务中都有涉及,例如:自动驾驶中的交通指示牌识别、交通违法管理中的车牌识别等。
[0003]相关技术中的字符识别流程一般是先检测后识别,即先检测出字符区域,再针对字符区域进行识别。但是,实际应用时,由于道路交通环境的复杂性,往往会检测到多个字符区域,这些字符区域可能来自不同类型的目标物上,例如,有的字符区域来自路牌,有的字符区域来自车辆车牌。为了实现识别不同类型的字符区域,通用的字符识别模型的数据量通常非常庞大,模型执行识别任务时将会非常消耗计算资源,使得模型的部署对终端设备的计算资源要求很高。

技术实现思路

[0004]本申请实施例的主要目的在于提出一种面向复杂交通场景的字符识别方法、装置、电子设备和计算机可读存储介质,旨在节省面向复杂交通场景进行字符识别所需的计算资源。
[0005]为实现上述目的,本申请实施例的第一方面提出了一种面向复杂交通场景的字符识别方法,所述方法包括:获取交通场景图像;通过预先训练好的检测模型对所述交通场景图像进行目标检测,获得包含待识别字符的目标框;通过预先训练好的分类模型对所述目标框进行特征提取和分类,获得第一特征图和目标类型;根据所述目标类型对所述第一特征图进行切分,得到字符特征图序列;通过预先训练好的识别模型对所述字符特征图序列进行处理,得到字符识别结果。
[0006]根据本申请一些实施例提供的字符识别方法,所述分类模型包括叠加设置的t层第一瓶颈网络和第一全连接网络,t为大于或等于3的整数;所述通过预先训练好的分类模型对所述目标框进行特征提取和分类,获得第一特征图和目标类型,包括:将所述目标框输入至第1层第一瓶颈网络;将第t

1层第一瓶颈网络和第t层第一瓶颈网络的输出结果相加,得到第一特征图;通过所述第一全连接网络对所述第一特征图进行处理,得到所述目标类型。
[0007]根据本申请一些实施例提供的字符识别方法,所述根据所述目标类型对所述第一特征图进行切分,得到字符特征图序列,包括:根据所述目标类型确定所述第一特征图包含的字符行数;当所述目标类型指示所述第一特征图包含多行字符,根据所述字符行数对所述第一特征图进行横向切分,得到多个第二特征图,每个第二特征图对应一行字符;按照切分顺序对所述多个第二特征图进行水平方向拼接,得到第三特征图;按照列对所述第三特
征图进行纵向切分,得到多个字符特征图,每个所述字符特征图中包含一个字符;按照切分顺序将所述多个字符特征图进行拼接,得到字符特征图序列。
[0008]根据本申请一些实施例提供的字符识别方法,在所述按照切分顺序对所述多个第二特征图进行水平方向拼接之前,所述方法还包括:对每个所述第二特征图进行预处理,将每个所述第二特征图调整为预设尺寸的图像。
[0009]根据本申请一些实施例提供的字符识别方法,所述识别模型包括叠加设置的m层第二瓶颈网络和第二全连接网络,m为大于或等于3的整数;所述通过预先训练好的识别模型对字符特征图序列进行处理,得到字符识别结果,包括:将所述字符特征图序列输入至第1层第二瓶颈网络;将第m

1层第二瓶颈网络和第m层第二瓶颈网络的输出结果相加,得到字符特征图序列特征;通过所述第二全连接网络对所述字符特征图序列特征进行处理,得到所述字符识别结果。
[0010]根据本申请一些实施例提供的字符识别方法,所述检测模型包括叠加设置的n层第三瓶颈网络和第三全连接网络,n为大于或等于3的整数;所述通过预先训练好的检测模型对所述交通场景图像进行目标检测,获得包含待识别字符的目标框,包括:将所述交通场景图像输入至第1层第三瓶颈网络;将第n

1层第三瓶颈网络和第n层第三瓶颈网络的输出结果相加,得到图像特征;通过所述第三全连接网络对所述图像特征进行处理,得到包含待识别字符的目标框。
[0011]根据本申请一些实施例提供的字符识别方法,所述获取交通场景图像,包括:获取描绘交通图像场景的视频,对所述视频进行抽帧处理,得到所述交通场景图像;以及对所述交通场景图像进行预处理,将所述交通场景图像调整为预设尺寸的图像。
[0012]为实现上述目的,本申请实施例的第二方面提出了一种面向复杂交通场景的字符识别装置,所述装置包括:
[0013]获取模块,用于获取交通场景图像;
[0014]检测模块,用于通过预先训练好的检测模型对所述交通场景图像进行目标检测,获得包含待识别字符的目标框;
[0015]分类模块,用于通过预先训练好的分类模型对所述目标框进行特征提取和分类,获得第一特征图和目标类型;
[0016]切分模块,用于根据所述目标类型对所述第一特征图进行切分,得到字符特征图序列;
[0017]识别模块,用于通过预先训练好的识别模型对所述字符特征图序列进行处理,得到字符识别结果。
[0018]为实现上述目的,本申请实施例的第三方面提出了一种电子设备,包括:至少一个处理器;
[0019]以及,与所述至少一个处理器通信连接的存储器;
[0020]其中,所述存储器存储有计算机程序,所述计算机程序被所述至少一个处理器执行,以使所述至少一个处理器能够执行如上第一方面任一项实施例所述的面向复杂交通场景的字符识别方法。
[0021]为实现上述目的,本申请实施例的第四方面提出了一种计算机可读存储介质,存储有计算机程序,所述计算机程序被处理器执行时实现如上第一方面任一项实施例所述的
面向复杂交通场景的字符识别方法。
[0022]本申请实施例提出了一种面向复杂交通场景的字符识别方法、装置、电子设备和计算机可读存储介质,获取交通场景图像,通过预先训练好的检测模型对所述交通场景图像进行目标检测,获得包含待识别字符的目标框,再通过预先训练好的分类模型对所述目标框进行特征提取和分类,获得第一特征图和目标类型,根据所述目标类型对所述第一特征图进行切分,得到字符特征图序列,最后通过预先训练好的识别模型对字符特征图序列进行处理,得到字符识别结果。本申请实施例的方案,通过分类模型对检测到的目标框进行特征提取,生成特征图,然后复用该特征图进行目标分类和字符识别,这样能够减小模型的处理数据量;以及,根据目标类型可以判断特征图中的字符行数,进而根据字符行数可以更加简单高效地完成字符分割。利用本申请提供的字符识别方法,可以很好地完成面向复杂交通场景的字符识别,特别地,可以简单高效地实现对于包含多行字符的目标框的识别,而且识别过程所需处理的数据量少,提升了计算速度,且节省了计算资源。
附图说明
[0023]图1是本申请实施例的提本文档来自技高网
...

【技术保护点】

【技术特征摘要】
1.一种面向复杂交通场景的字符识别方法,其特征在于,方法包括:获取交通场景图像;通过预先训练好的检测模型对所述交通场景图像进行目标检测,获得包含待识别字符的目标框;通过预先训练好的分类模型对所述目标框进行特征提取和分类,获得第一特征图和目标类型;根据所述目标类型对所述第一特征图进行切分,得到字符特征图序列;通过预先训练好的识别模型对所述字符特征图序列进行处理,得到字符识别结果。2.根据权利要求1所述的方法,其特征在于,所述分类模型包括叠加设置的t层第一瓶颈网络和第一全连接网络,t为大于或等于3的整数;所述通过预先训练好的分类模型对所述目标框进行特征提取和分类,获得第一特征图和目标类型,包括:将所述目标框输入至第1层第一瓶颈网络;将第t

1层第一瓶颈网络和第t层第一瓶颈网络的输出结果相加,得到所述第一特征图;通过所述第一全连接网络对所述第一特征图进行处理,得到所述目标类型。3.根据权利要求1所述的方法,其特征在于,所述根据所述目标类型对所述第一特征图进行切分,得到字符特征图序列,包括:根据所述目标类型确定所述第一特征图包含的字符行数;当所述目标类型指示所述第一特征图包含多行字符,根据所述字符行数对所述第一特征图进行横向切分,得到多个第二特征图,每个第二特征图对应一行字符;按照切分顺序对所述多个第二特征图进行水平方向拼接,得到第三特征图;按照列对所述第三特征图进行纵向切分,得到多个字符特征图,每个所述字符特征图中包含一个字符;按照切分顺序将所述多个字符特征图进行拼接,得到字符特征图序列。4.根据权利要求3所述的方法,其特征在于,在所述按照切分顺序对所述多个第二特征图进行水平方向拼接之前,所述方法还包括:对每个所述第二特征图进行预处理,将每个所述第二特征图调整为预设尺寸的图像。5.根据权利要求1所述的方法,其特征在于,所述识别模型包括叠加设置的m层第二瓶颈网络和第二全连接网络,m为大于或等于3的整数;所述通过预先训练好的识别模型对所述字符特征图序列进行处理,得到字符识别结果,包括:将所述字符特征图序列输入至第...

【专利技术属性】
技术研发人员:唐子豪刘莉红刘玉宇
申请(专利权)人:平安科技深圳有限公司
类型:发明
国别省市:

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

1