[发明专利]文档方向识别方法、装置及存储介质

权利要求书

1.一种文档方向识别方法，其特征在于，包括：

获取文档图像，所述文档图像中包括待识别文档；

确定所述文档图像对应的二值图像，并从所述二值图像中选取多个图像块；

确定每个所述图像块的信息含量值；

根据所述信息含量值和所述图像块识别所述待处理文档在所述文档图像中的方向。

2.根据权利要求1所述的文档方向识别方法，其特征在于，所述确定每个所述图像块的信息含量值，包括:

利用已训练的全卷积网络模型确定每个所述图像块对应的热力图像；

确定每个所述热力图像中各像素点的第一像素值；

根据所述第一像素值确定对应热力图像的第一像素平均值；

根据所述第一像素平均值确定对应图像块的信息含量值。

3.根据权利要求2所述的文档方向识别方法，其特征在于，所述根据所述第一像素平均值确定对应图像块的信息含量值，包括：

确定每个所述图像块中的连通域、以及每个所述连通域的区域面积；

根据所述区域面积从所述连通域中确定目标连通域，并获取所述目标连通域的数量；

根据所述第一像素平均值、以及所述目标连通域的数量确定对应图像块的信息含量值。

4.根据权利要求3所述的文档方向识别方法，其特征在于，所述根据所述第一像素平均值、以及所述目标连通域的数量确定对应图像块的信息含量值，包括：

根据所述第二像素值确定对应图像块的第二像素平均值；

根据所述第二像素平均值和第二像素值确定对应图像块的像素标准差值；

根据同一图像块对应的第一像素平均值、第二像素平均值、像素标准差值以及所述目标连通域的数量，确定对应图像块的信息含量值。

5.根据权利要求1-4中任意一项所述的文档方向识别方法，其特征在于，所述根据所述信息含量值和所述图像块识别所述待处理文档在所述文档图像中的方向，包括：

根据所述信息含量值从所述多个图像块中确定目标图像块；

利用已训练的网络模型确定每个所述目标图像块对应的阅读方向角；

根据所述阅读方向角和信息含量值识别所述待处理文档在所述文档图像中的方向。

6.根据权利要求5所述的文档方向识别方法，其特征在于，所述根据所述阅读方向角和信息含量值识别所述待处理文档在所述文档图像中的方向，包括：

将所述信息含量值进行求和，得到和值；

根据所述和值和信息含量值确定每个所述目标图像块的权重值；

根据所述权重值对所述阅读方向角进行加权计算，得到目标方向角，并将所述目标方向角指示的方向作为所述待处理文档在所述文档图像中的方向。

7.根据权利要求5所述的文档方向识别方法，其特征在于，所述网络模型包括残差网络模型和注意力模型，所述利用已训练的网络模型确定每个所述目标图像块对应的阅读方向角，包括：

将每个所述目标图像块输入已训练的残差网络模型中，以对所述目标图像块进行处理，所述残差网络模型包括至少四层残差块；

在处理过程中，获取输入第二层残差块中的第一特征图；

将获取的所述第一特征图输入已训练的注意力模型中，得到注意力图像；

获取从第三层残差块中输出的第二特征图；

对所述第二特征图和注意力图进行点乘运算，得到第三特征图；

将所述第三特征图输入第四层残差块中，以得到阅读方向角。

8.一种文档方向识别装置，其特征在于，包括：

获取模块，用于获取文档图像，所述文档图像中包括待识别文档；

第一确定模块，用于确定所述文档图像对应的二值图像，并从所述二值图像中选取多个图像块；

第二确定模块，用于确定每个所述图像块的信息含量值；

识别模块，用于根据所述信息含量值和所述图像块识别所述待处理文档在所述文档图像中的方向。