[发明专利]一种电子文件的旋转校正方法和装置

权利要求书

1.一种电子文件的旋转校正方法，其特征在于，所述方法包括：

获取待校正的目标电子文件图像；

将所述目标电子文件图像输入角度识别模型，所述角度识别模型包括特征提取层、特征融合层、特征转换层和分类层；

其中，所述特征提取层用于基于若干不同的特征提取方向对所述目标电子文件图像进行特征提取，得到各个特征提取方向对应的初始特征；

所述特征融合层用于将各个特征提取方向对应的初始特征进行融合，得到综合特征；

所述特征转换层用于将所述综合特征转换为若干个标准旋转角度对应的概率值，所述标准旋转角度的数量与角度识别精度正相关；

所述分类层用于基于各个标准旋转角度对应的概率值确定所述目标电子文件图像的校正旋转角度；

基于所述角度识别模型输出的校正旋转角度对所述目标电子文件图像进行校正。

2.根据权利要求1所述方法，其特征在于，

所述特征提取层包括若干个特征提取模块，每个特征提取模块执行所述基于若干不同的特征提取方向对所述目标电子文件图像进行特征提取，得到与所述特征提取模块对应的各个特征提取方向的初始特征；

所述特征融合层包括第一子特征融合层、子池化层和第二子特征融合层，其中，所述第一子特征融合层用于针对每个特征提取模块对应的各个特征提取方向的初始特征，将所述各个特征提取方向的初始特征进行融合，得到所述特征提取模块对应的中间特征；

所述子池化层用于分别对每个特征提取模块对应的中间特征进行池化处理，得到每个特征提取模块对应的池化处理后的中间特征；

所述第二子特征融合层用于对各个特征提取模块对应的池化处理后的中间特征进行融合，得到所述目标电子文件图像的综合特征。

3.根据权利要求1或2所述方法，其特征在于，所述基于若干不同的特征提取方向对所述目标电子文件图像进行特征提取，得到各个特征提取方向对应的初始特征，包括：

分别采用第一特征提取方向对应的第一卷积核和第二特征提取方向对应的第二卷积核对所述目标电子文件进行特征提取，得到第一特征提取方向对应的初始特征和第二特征提取方向对应的初始特征，其中，所述第一卷积核和所述第二卷积核互为转置矩阵。

4.根据权利要求1所述方法，其特征在于，特征转换层将所述综合特征转换为若干个标准旋转角度对应的概率值，包括：

基于reshape函数将所述综合特征转换为旋转角度向量，其中，所述旋转角度向量中元素的数量与所述标准旋转角度的数量相同，所述旋转角度向量的每个元素值分别代表对应标准旋转角度的概率值。

5.根据权利要求1所述方法，其特征在于，所述基于各个标准旋转角度对应的概率值，确定所述目标电子文件图像的校正旋转角度，包括：

将概率值高的标准旋转角度确定为所述目标电子文件图像的校正旋转角度。

6.根据权利要求1所述方法，其特征在于，所述待校正的目标电子文件图像通过扫描纸质文件得到。

7.一种电子文件的旋转校正装置，其特征在于，所述装置包括：

获取单元，用于获取待校正的目标电子文件图像；

输入单元，用于将所述目标电子文件图像输入角度识别模型，所述角度识别模型包括特征提取层、特征融合层、特征转换层和分类层；

其中，所述特征提取层用于基于若干不同的特征提取方向对所述目标电子文件图像进行特征提取，得到各个特征提取方向对应的初始特征；

所述特征融合层用于将各个特征提取方向对应的初始特征进行融合，得到综合特征；

所述特征转换层用于将所述综合特征转换为若干个标准旋转角度对应的概率值，所述标准旋转角度的数量与角度识别精度正相关；

所述分类层用于基于各个标准旋转角度对应的概率值确定所述目标电子文件图像的校正旋转角度；

校正单元，用于基于所述角度识别模型输出的校正旋转角度对所述目标电子文件图像进行校正。