畸变文本图像矫正模型训练方法及畸变文本图像矫正方法

本发明公开了一种畸变文本图像矫正模型训练方法及畸变文本图像矫正方法，应用于畸变文本图像矫正领域，基于自注意力机制提取畸变文本图像样本的全局特征；从全局特征中提取局部特征，将局部特征进行特征融合；将融合特征经全连接层得到仿射变换矩阵，将仿射变换矩阵与预设矩阵相乘得到索引矩阵；基于索引矩阵对畸变文本图像样本进行像素重排得到畸变矫正后的畸变矫正图像；基于畸变矫正图像更新模型参数得到训练完成的畸变文本图像矫正模型。通过全局和局部特征提取的方式以构建更加健全的仿射变换矩阵，促进像素重排的准确性，有效矫正现实场景中多种图像的畸变，且基于自注意力的并行计算特性，能够在收集全局信息的基础上，快速完成计算。