[发明专利]基于DB与CRNN的特种作业证件检测方法及系统

说明书

本发明公开了一种基于DB与CRNN的特种作业证件检测方法及系统，该方法包括：将每张目标特种作业证件图像输入到DB文本检测网络模型，以确定每张目标特种作业证件图像对应的文本框数据集合；将每张目标特种作业证件图像以及每张目标特种作业证件图像对应的文本框数据集合输入到CRNN文本识别网络模型中，以确定每张所述目标特种作业证件图像中每个所述目标文本框内的文本信息。DB文本检测网络中的Backbone模块采用的是MobileNetV3‑large结构；CRNN文本识别网络中的CNN模块的部分结构采用的是MobileNetV3‑small结构。本发明能够达到降低人工工作量并提高证件图像检测效率的目的。

技术领域

本发明涉及光学字符识别技术领域，特别是涉及一种基于DB与CRNN的特种作业证件检测方法及系统。

背景技术

在5G基站施工过程中，施工人员具有合格有效的特种作业证件，是施工过程中不可或缺的安全保障。目前关于特种作业证件的检测多是以人工方式进行的，检测效率低下，且不能及时有效地得到特种作业证件检测的反馈。

发明内容

本发明的目的是提供一种基于DB与CRNN的特种作业证件检测方法及系统，以达到降低人工工作量并提高证件图像检测效率的目的。

为实现上述目的，本发明提供了如下方案：

一种基于DB与CRNN的特种作业证件检测方法，包括：

获取特种作业证件图像数据集；所述特种作业证件图像数据集包括多张目标特种作业证件图像，且每张所述目标特种作业证件图像均有文本信息；将每张所述目标特种作业证件图像输入到DB文本检测网络模型，以确定每张所述目标特种作业证件图像对应的文本框数据集合；所述文本框数据集合中的元素表示目标文本框的位置信息；将每张所述目标特种作业证件图像以及每张所述目标特种作业证件图像对应的文本框数据集合输入到CRNN文本识别网络模型中，以确定每张所述目标特种作业证件图像中每个所述目标文本框内的文本信息；所述文本信息包括施工人员姓名、施工人员性别、证件号、作业类别以及证件有效日期中的至少一种；

其中，所述DB文本检测网络模型是基于DB文本检测网络和第一训练数据集训练得到的；所述DB文本检测网络中的Backbone模块采用的是MobileNetV3-large结构；所述第一训练数据集中的每个元素均包括历史特种作业证件图像以及所述历史特种作业证件图像对应的第一类别标签；所述第一类别标签为历史文本框的位置信息；所述CRNN文本识别网络模型是基于CRNN文本识别网络和第二训练数据集训练得到的；所述CRNN文本识别网络中的CNN模块的部分结构采用的是MobileNetV3-small结构；所述第二训练数据集中的每个元素均包括历史特种作业证件图像以及所述历史特种作业证件图像对应的第二类别标签；所述第二类别标签为历史文本信息。

一种基于DB与CRNN的特种作业证件检测系统，包括：

数据获取模块，用于获取特种作业证件图像数据集；所述特种作业证件图像数据集包括多张目标特种作业证件图像，且每张所述目标特种作业证件图像均有文本信息；文本框数据集合确定模块，用于将每张所述目标特种作业证件图像输入到DB文本检测网络模型，以确定每张所述目标特种作业证件图像对应的文本框数据集合；所述文本框数据集合中的元素表示目标文本框的位置信息；文本信息确定模块，用于将每张所述目标特种作业证件图像以及每张所述目标特种作业证件图像对应的文本框数据集合输入到CRNN文本识别网络模型中，以确定每张所述目标特种作业证件图像中每个所述目标文本框内的文本信息；所述文本信息包括施工人员姓名、施工人员性别、证件号、作业类别以及证件有效日期中的至少一种；