[发明专利]文档异常检测网络模型构建方法、装置、电子设备及介质

权利要求书

1.一种文档异常检测网络模型构建方法，其特征在于，所述方法包括：

基于正常文档图像随机选取文字区域，根据所述文字区域生成文档异常图像样本集，所述文档异常图像样本集包括多个文档异常图像样本；

对各所述文档异常图像进行文档异常打标，得到多个打标后图像样本，生成各所述打标后图像样本对应的打标信息文件；

确定文档异常图像训练集的第一样本数量，从多个所述打标信息文件中抽取所述第一样本数量的打标信息文件，根据所述第一样本数量的打标信息文件生成训练图像索引列表；

基于YOLO框架构建初始文档异常检测网络模型，根据真实边界框的尺寸、所述训练图像索引列表、所述训练图像索引列表对应的文档异常训练图像对所述初始文档异常检测网络模型进行训练，得到文档异常检测网络模型；

所述根据真实边界框的尺寸、所述训练图像索引列表、所述训练图像索引列表对应的文档异常图像对所述初始文档异常检测网络模型进行训练，包括：

通过所述初始文档异常检测网络模型加载所述训练图像索引列表对应的打标信息文件，获取加载到的打标信息文件的文档异常坐标信息，将所述真实边界框的尺寸信息作为K-means聚类的输入数据；

根据所述文档异常坐标信息、所述真实边界框的尺寸信息和所述文档异常训练图像对所述初始文档异常检测网络模型进行训练。

2.根据权利要求1所述的方法，其特征在于，所述根据所述文字区域生成文档异常图像样本，包括：

通过OCR确定所述文字区域的文字位置，并获取所述文字位置对应的文本内容集合；

通过OPENCV图像处理算法获取所述文本内容集合在所述正常文档图像的背景和字体颜色，通过所述文字位置的宽高及文本行数计算字号，根据所述背景、所述字体颜色、所述字号构建边缘框；

根据所述边缘框、所述正常文档图像的原文字框生成文档异常图像样本。

3.根据权利要求2所述的方法，其特征在于，所述文档异常图像样本包括文本重叠样本和/或文本遮挡样本，所述根据所述边缘框、所述正常文档图像的原文字框生成文档异常图像样本，包括：

通过计算所述边缘框和原文字框的交并比构建所述文本重叠样本；和/或，

根据所述文本内容集合通过预设文本框对所述正常文档图像进行遮挡，得到所述文本遮挡样本。

4.根据权利要求1所述的方法，其特征在于，所述方法还包括：

将各所述文档异常图像存入图像文件夹；

将各所述打标信息文件存入打标文件夹，其中，所述打标文件夹下的各打标信息文件与所述图像文件夹下的各文档异常图像一一对应。

5.根据权利要求1所述的方法，其特征在于，所述打标信息文件包括自定义图像对象名、图像文件名、图像文件路径、图像尺寸和文档异常坐标信息。

6.根据权利要求1所述的方法，其特征在于，所述方法包括：

确定文档异常图像测试集的第二样本数量；

从多个所述打标信息文件中抽取所述第二样本数量的打标信息文件，根据所述第二样本数量的打标信息文件生成测试图像索引列表；

通过所述文档异常检测网络模型根据所述测试图像索引列表对应的文档异常坐标信息和文档异常测试图像确定文档异常结果的误检率和漏检率。

7.一种文档图像的文档异常检测方法，其特征在于，所述方法包括：

向文档异常检测网络模型输入待检测文档图像，所述文档异常检测网络模型根据权利要求1-6中任一项所述的文档异常检测网络模型构建方法获取得到；

通过所述文档异常检测网络模型对所述待检测文档图像进行检测，得到文档异常输出结果，所述文档异常输出结果包括文档异常坐标信息、对象置信度、类别概率和所属类别；

根据文档异常坐标信息、对象置信度、类别概率和所属类别生成文档异常检测结果。