[发明专利]一种图像矫正方法、装置、电子设备及存储介质

权利要求书

1.一种图像矫正方法，其特征在于，包括：

对待矫正图像进行语义分割，得到所述待矫正图像对应的第一掩膜图像；

获取所述第一掩膜图像的前景像素点特征；

对所述前景像素点特征进行主成分分析，得到主成分向量；

根据所述主成分向量确定所述待矫正图像对应的旋转角度；

根据所述旋转角度对所述待矫正图像进行旋转，得到矫正后图像。

2.根据权利要求1所述的方法，其特征在于，所述获取所述第一掩膜图像的前景像素点，包括：

对所述第一掩膜图像进行预设采样率的下采样，得到第二掩膜图像；

遍历所述第二掩膜图像的前景像素点，根据前景像素点坐标得到所述前景像素点特征。

3.根据权利要求1或2所述的方法，其特征在于，所述对所述前景像素点特征进行主成分分析，得到主成分向量，包括：

根据所述前景像素点特征中的前景像素点坐标生成协方差矩阵；

计算所述协方差矩阵的特征向量及所述特征向量对应的特征值；

将所述特征值符合第一预设条件的特征向量作为所述主成分向量。

4.根据权利要求1所述的方法，其特征在于，所述根据所述主成分向量确定所述待矫正图像对应的旋转角度，包括：

获取基准方向；

计算所述主成分向量与所述基准方向之间的夹角；

根据所述夹角确定所述旋转角度。

5.根据权利要求1所述的方法，其特征在于，所述对待矫正图像进行语义分割，得到所述待矫正图像对应的第一掩膜图像，包括：

识别所述待矫正图像中的目标物；

当所述目标物的个数大于1时，对所述待矫正图像进行裁剪，得到每个所述目标物对应的子图像；

对所述子图像进行语义分割，得到每个所述目标物对应的第一掩膜图像。

6.根据权利要求1所述的方法，其特征在于，所述获取所述第一掩膜图像的前景像素点特征，包括：

当所述第一掩膜图像中包括未连通的至少两个目标区域，或目标区域的连通部分符合第二预设条件时，对所述第一掩膜图像进行裁剪，得到包括每个所述目标区域的第三掩膜图像；

获取每张所述第三掩膜图像的前景像素点。

7.根据权利要求1所述的方法，其特征在于，所述根据所述旋转角度对所述待矫正图像进行旋转，得到矫正后图像，包括：

获取旋转中心点坐标、缩放因子及所述旋转角度；

根据所述旋转中心点坐标、所述缩放因子及所述旋转角度生成所述待矫正图像对应的旋转矩阵；

根据所述旋转矩阵对所述待矫正图像进行旋转，得到矫正后图像。

8.一种图像矫正装置，其特征在于，包括：

语义分割模块，用于对待矫正图像进行语义分割，得到所述待矫正图像对应的第一掩膜图像；

获取模块，用于获取所述第一掩膜图像的前景像素点特征；

主成分分析模块，用于对所述前景像素点特征进行主成分分析，得到主成分向量；

计算模块，用于根据所述主成分向量确定所述待矫正图像对应的旋转角度；

旋转模块，用于根据所述旋转角度对所述待矫正图像进行旋转，得到矫正后图像。

9.一种电子设备，其特征在于，包括：处理器、通信接口、存储器和通信总线，其中，处理器，通信接口，存储器通过通信总线完成相互间的通信；

所述存储器，用于存放计算机程序；

所述处理器，用于执行所述计算机程序时，实现权利要求1-7任一项所述的方法步骤。

10.一种计算机可读存储介质，其上存储有计算机程序，其特征在于，该计算机程序被处理器执行时实现权利要求1-7任一项所述的方法步骤。