[发明专利]文档矫正方法、装置、电子设备和存储介质

权利要求书

1.一种文档矫正方法，包括：

对包括待矫正文档的原始图像进行语义分割，得到边缘像素点；

基于所述边缘像素点，确定第一边缘线；

根据矫正后文档对应的图像尺寸，确定目标图像的网格图像；

对所述第一边缘线和所述网格图像的第二边缘线进行等分处理，分别得到所述第一边缘线的第一关键点和所述第二边缘线中与所述第一关键点对应的第二关键点；以及

根据所述第一关键点和所述第二关键点之间的对应关系，生成所述目标图像；

其中，所述边缘像素点对应至少两个预定类别；所述预定类别与边缘线相对于文档所在的方位相关；所述基于所述边缘像素点，确定第一边缘线，包括：

响应于根据所述边缘像素点的类别确定所述待矫正文档是双页文档，确定所述边缘像素点中属于所述双页文档中每个的单页文档的边缘点；以及

基于所述单页文档的边缘点，确定所述单页文档的所述第一边缘线。

2.根据权利要求1所述的方法，其中，所述预定类别包括：上边缘类别、下边缘类别、中边缘类别、左边缘类别和右边缘类别；

所述方法还包括：

响应于确定所述边缘像素点的类别包括所述中边缘类别或者所述边缘像素点的类别包括所述预定类别的全部类别，确定所述待矫正文档为双页文档。

3.根据权利要求1～2中任一项所述的方法，其中，所述对包括待矫正文档的原始图像进行语义分割，得到边缘像素点包括：

采用语义分割模型对所述原始图像进行处理，得到所述原始图像中每个像素点属于至少两个类别中每个类别的概率值；以及

基于所述概率值，确定所述边缘像素点，

其中，所述至少两个类别包括：背景类别和至少两个预定类别。

4.根据权利要求1所述的方法，其中，所述基于所述边缘像素点，确定第一边缘线，还包括：

响应于根据所述边缘像素点的类别确定所述待矫正文档不是所述双页文档，确定所述待矫正文档为单页文档，并确定所述边缘像素点中属于所述单页文档的边缘点。

5.根据权利要求1或4所述的方法，其中，所述基于所述单页文档的边缘点，确定所述单页文档的所述第一边缘线，包括：

基于所述单页文档的边缘点，采用多项式拟合算法确定所述第一边缘线。

6.根据权利要求5所述的方法，其中，所述基于所述单页文档的边缘点，采用多项式拟合算法确定所述第一边缘线包括：

基于所述单页文档的边缘点，采用所述多项式拟合算法确定分别针对至少两个预定阶次的至少两个边缘线拟合数据；以及

根据所述边缘线拟合数据中与所述单页文档的边缘点之间的差异最小的目标拟合数据，确定所述第一边缘线。

7.根据权利要求6所述的方法，其中，所述根据所述边缘线拟合数据中与所述单页文档的边缘点之间的差异最小的目标拟合数据，确定所述第一边缘线包括：

根据所述目标拟合数据，确定所述单页文档的顶点位置；以及

根据所述顶点位置和所述目标拟合数据，确定经过所述顶点位置且围成封闭区域的边缘线，得到所述第一边缘线。

8.根据权利要求1或4所述的方法，其中，所述单页文档的边缘点包括至少两个目标类别的至少两组边缘点；所述目标类别包括：所述预定类别中与边缘线相对于文档所在的四个方位相对应的四个类别；

所述基于所述单页文档的边缘点，确定所述单页文档的第一边缘线包括：

基于每个所述目标类别的一组边缘点，确定每个所述目标类别的参考边缘点；以及

基于所述参考边缘点，确定所述待矫正文档在每个所述目标类别对应的方位的边缘线。

9.根据权利要求8所述的方法，其中，所述基于每个所述目标类别的一组边缘点，确定每个所述目标类别的参考边缘点包括：

对于每个所述目标类别的一组边缘点中垂直于目标方向排列的至少两个边缘点，确定至少两个所述边缘点的坐标值的平均值所指示的点为所述参考边缘点，

其中，所述目标方向为每个所述目标类别所对应的方位所在方向。