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

权利要求书

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

获取特种作业证件图像数据集；所述特种作业证件图像数据集包括多张目标特种作业证件图像，且每张所述目标特种作业证件图像均有文本信息；

将每张所述目标特种作业证件图像输入到DB文本检测网络模型，以确定每张所述目标特种作业证件图像对应的文本框数据集合；所述文本框数据集合中的元素表示目标文本框的位置信息；

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

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

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

2.根据权利要求1所述的一种基于DB与CRNN的特种作业证件检测方法，其特征在于，还包括：基于所述文本信息，确定每张所述特种作业证件是否符合施工作业要求。

3.根据权利要求1所述的一种基于DB与CRNN的特种作业证件检测方法，其特征在于，所述将每张所述目标特种作业证件图像输入到DB文本检测网络模型，以确定每张所述目标特种作业证件图像对应的文本框数据集合，具体包括：

对每张所述目标特种作业证件图像进行预处理；所述预处理包括：解码、归一化、重新排列以及图像缩放；

将每张预处理后的目标特种作业证件图像输入到DB文本检测网络模型，以确定每张所述目标特种作业证件图像对应的文本框数据集合。

4.根据权利要求3所述的一种基于DB与CRNN的特种作业证件检测方法，其特征在于，所述将每张所述目标特种作业证件图像以及每张所述目标特种作业证件图像对应的文本框数据集合输入到CRNN文本识别网络模型中，以确定每张所述目标特种作业证件图像中每个所述目标文本框内的文本信息，具体包括：

将每张预处理后的目标特种作业证件图像以及每张所述预处理后的目标特种作业证件图像对应的预处理后的文本框数据集合输入到CRNN文本识别网络模型中，以确定每张所述目标特种作业证件图像中每个所述目标文本框内的文本信息。

5.根据权利要求1所述的一种基于DB与CRNN的特种作业证件检测方法，其特征在于，所述DB文本检测网络模型的确定过程为：

构建DB文本检测网络；

确定第一训练数据集；

基于所述第一训练数据集，训练所述DB文本检测网络，以得到DB文本检测网络模型。

6.根据权利要求5所述的一种基于DB与CRNN的特种作业证件检测方法，其特征在于，所述确定第一训练数据集，具体包括：

获取原始证件图像数据集；所述原始证件图像数据集包括多张原始历史特种作业证件图像；

采用半自动标注工具，对每张所述原始历史特种作业证件图像进行标注，得到每张历史标注图像以及每张所述历史标注图像对应的第一类别标签；

对所述每张历史标注图像均进行预处理，以得到历史特种作业证件图像；解码、归一化、重新排列以及图像缩放；所述历史特种作业证件图像对应的第一类别标签为所述历史标注图像对应的第一类别标签。