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

权利要求书

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

获取需要进行图像校正的待矫正图像；

获取预训练的第一矫正模型和第二矫正模型；

将所述待矫正图像输入所述第一矫正模型进行图像矫正，得到候选矫正图像；

根据所述候选矫正图像与所述待矫正图像每一像素点之间的差异，获取所述待矫正图像每一像素点对应的像素矫正增益；

获取所述待矫正图像所有像素点的像素矫正增益的平均值，作为所述待矫正图像的第一矫正增益，并根据所述第一矫正增益矫正所述待矫正图像得到第一矫正图像；

根据所述第二矫正模型获取对应所述第一矫正图像的第二矫正增益，并根据所述第二矫正增益矫正所述第一矫正图像得到矫正结果图像。

2.根据权利要求1所述的图像矫正方法，其特征在于，所述第一矫正模型与所述第二矫正模型结构相同，且互不共享参数，所述第一矫正模型包括编码模块、自注意力模块以及解码模块，所述将所述待矫正图像输入所述第一矫正模型进行图像矫正，得到候选矫正图像，包括：

基于所述编码模块对所述待矫正图像进行编码，得到特征图；

基于所述自注意力模块对所述特征图进行自注意处理，得到自注意图；

基于所述解码模块对所述自注意图进行解码，得到所述候选矫正图像。

3.根据权利要求2所述的图像矫正方法，其特征在于，所述自注意力模块包括第一卷积模块、第二卷积模块以及第三卷积模块，所述基于所述自注意力模块对所述特征图进行自注意处理，得到自注意图，包括：

基于所述第一卷积模块对所述特征图进行卷积处理，得到第一卷积特征图，以及基于所述第二卷积模块对所述特征图进行卷积处理，得到第二卷积特征图；

将所述第一卷积特征图转置后与所述第二卷积特征图相乘得到融合图，并对所述融合图进行归一化处理，得到注意力图；

基于所述第三卷积模块对所述特征图进行卷积处理，得到第三卷积特征图，并将所述第三卷积特征图和所述注意力图相乘得到所述自注意图。

4.根据权利要求1-3任一项所述的图像矫正方法，其特征在于，所述获取需要进行图像矫正的待矫正图像之前，还包括：

获取样本图像及其对应的目标矫正增益；

根据所述样本图像及其对应的目标矫正增益对所述第一矫正模型和所述第二矫正模型进行训练。

5.根据权利要求4所述的图像矫正方法，其特征在于，所述根据所述样本图像及其对应的目标矫正增益对所述第一矫正模型和所述第二矫正模型进行训练，包括：

根据所述第一矫正模型获取对应所述样本图像的第三矫正增益，并根据所述第三矫正增益矫正所述样本图像得到第一样本矫正图像；

获取所述第三矫正增益与所述目标矫正增益的第一矫正差异，并将所述目标矫正增益与所述第三矫正增益相除得到次级目标矫正增益；

根据所述第二矫正模型获取对应所述第一样本矫正图像的第四矫正增益，并获取所述第四矫正增益与所述次级目标矫正增益的第二矫正差异；

融合所述第一矫正差异与所述第二矫正差异得到融合差异，并根据所述融合差异对所述第一矫正模型和/或所述第二矫正模型的参数进行调整。

6.根据权利要求5所述的图像矫正方法，其特征在于，按照如下公式融合得到所述融合差异：

L_total＝w*L_f+(1-w)L_s；

其中，L_total表示所述融合差异，w表示占比系数，L_f表示所述第一矫正差异，L_s表示所述第二矫正差异。