[发明专利]一种图像倾斜的校正方法及装置

权利要求书

1.一种图像倾斜的校正方法，其特征在于：包括以下步骤：

步骤S1、形成关键角点预测网络模型；

步骤S2、利用关键角点预测网络模型对待校正的文档图像进行预测，生成待校正的文档图像的16个关键点；

步骤S3、对每条边上的6个关键点两两连接，并将连接后的曲线分别拟合得到4个边，延长所述4个边，得到4个纸张角点；

步骤S4、以4个纸张角点为原点，计算得到对应的纸张的最长的高度和宽度，以左上角点为基准，计算得到新的右上角点、右下角点和左下角点，然后得到变换后的4个目标角点，通过透视变换原理，可以得到透视矩阵，使用透视变换，得到校正后的图像。

2.根据权利要求1所述的一种图像倾斜的校正方法，其特征在于：所述的形成关键角点预测网络模型的步骤包括：

步骤S101、使用文档边角点标注方法，标注每一张要训练的样本文档图像，形成文档边角点标注的训练集；

步骤S102、在文档边角点标注的训练集上，构建基于约束条件的关键角点预测网络模型。

3.根据权利要求2所述的一种图像倾斜的校正方法，其特征在于：所述的文档边角点标注方法为：每张文档图像使用16个关键点进行标注，包括8个角点标注以及8个边内点的标注。

4.根据权利要求3所述的一种图像倾斜的校正方法，其特征在于：当角点没有被遮挡时，其2个角点标注的位置重合。

5.根据权利要求3所述的一种图像倾斜的校正方法，其特征在于：所述的8个边内点，由分别在4条边上各取的2个内点构成。

6.根据权利要求5所述的一种图像倾斜的校正方法，其特征在于：每条边上的两个内点，分别在边长度的三分之一和三分之二处。

7.根据权利要求2所述的一种图像倾斜的校正方法，其特征在于：所述的约束条件为：第二关键点、第三关键点、第四关键点和第五关键点在一条直线上；第六关键点、第七关键点、第八关键点和第九关键点在一条直线上；第十关键点、第十一关键点、第十二关键点和第十三关键点在一条直线上；第十四关键点、第十五关键点、第十六关键点和第一关键点在一条直线上。

8.一种图像倾斜的校正装置，其特征在于：包括获取模块、关键角点预测模块、纸张角点拟合模块和校正模块；获取模块的输入端获取待校正的文档图像；获取模块的输出端连接关键角点预测模块的输入端，关键角点预测模块的输出端连接纸张角点拟合模块的输入端；纸张角点拟合模块的输出端连接校正模块的输入端，校正模块的输出端输出校正后的文档图像。

9.根据权利要求8所述的获取模块：一种图像倾斜的校正装置，其特征在于：

获取模块：用于获取待校正的文档图像；

关键角点预测模块：存有关键角点预测模型，用于预测待校正的文档图像的16个关键点；

纸张角点拟合模块：根据文档图像的16个关键点，拟合生成文档图像的4个纸张角点；

校正模块：根据生成的4个纸张角点，计算得到4个目标角点，通过透视变换原理得到透视矩阵，使用透视变换，得到原文档图像的校正图像。