[发明专利]文本图像超分辨率重建方法及其相关设备

权利要求书

1.一种文本图像超分辨率重建方法，其特征在于，包括下述步骤：

接收低分辨率图片和对应的高分辨率图片，将所述低分辨率图片输入至预先训练的场景文本识别模型中，获得输出的文本位置信息和文本内容信息；

基于所述文本位置信息和所述文本内容信息生成文本掩模，并对所述文本掩膜进行上采样，获得目标掩膜；

将所述低分辨率图片和所述目标掩膜输入至预设的对抗网络的生成层中，获得输出的超分辨率图片；

将所述超分辨率图片和所述高分辨率图片同时输入至所述对抗网络的判别层中，获得输出的判别结果，并基于所述判别结果计算判别准确率；

基于所述低分辨率图片和所述目标掩膜计算所述对抗网络的损失函数，直至所述损失函数收敛，且所述判别准确率低于准确率阈值时，获得训练后的对抗网络；

接收待转化低分辨率图片，将所述待转化低分辨率图片输入至训练后的对抗网络中，获得输出的目标超分辨率图片。

2.根据权利要求1所述的文本图像超分辨率重建方法，其特征在于，所述基于所述文本位置信息和所述文本内容信息生成文本掩模的步骤包括：

基于所述文本内容信息修正所述文本位置信息，获得目标文本位置信息；

基于所述目标文本位置信息生成所述文本掩膜。

3.根据权利要求1所述的文本图像超分辨率重建方法，其特征在于，所述基于所述低分辨率图片和所述目标掩膜计算所述对抗网络的损失函数的步骤包括：

基于所述低分辨率图片计算所述对抗网络的内容损失函数，所述内容损失函数的特征为：

其中，为所述内容损失函数，为所述高分辨率图片在(x，y)位置的像素点的值，G_θG(I^LR)_x，y为所述超分辨率图片在(x，y)位置的像素点的值，rW和rH分别为所述超分辨率图片的宽和长，r²WH为所述超分辨率图片的像素点的总数量。

4.根据权利要求1所述的文本图像超分辨率重建方法，其特征在于，所述基于所述低分辨率图片和所述目标掩膜计算所述对抗网络的损失函数的步骤包括：

基于所述低分辨率图片计算所述对抗网络的对抗损失函数，所述对抗损失函数的特征为：

其中，为所述对抗损失函数，G_θG(I^LR)为所述超分辨率图片，D_θD为所述判别层，M为所述超分辨率图片的总数量，m表示所述超分辨率图片的个数。

5.根据权利要求1所述的文本图像超分辨率重建方法，其特征在于，所述基于所述低分辨率图片和所述目标掩膜计算所述对抗网络的损失函数的步骤包括：

基于所述低分辨率图片计算所述对抗网络的正则化损失函数，所述正则化损失函数的特征为：

其中，为所述正则化损失函数，G_θG(I^LR)_x，y为所述超分辨率图片在(x，y)位置的像素点的值，rW和rH分别为所述目标掩膜的宽和长，r²WH为所述目标掩膜中像素点的总数量，||||表示范数，表示梯度。

6.根据权利要求1所述的文本图像超分辨率重建方法，其特征在于，所述基于所述低分辨率图片和所述目标掩膜计算所述对抗网络的损失函数的步骤包括：

基于所述低分辨率图片计算所述对抗网络的文本感知损失函数，所述文本感知损失函数的特征为：

其中，其中，l^TR为所述文本感知损失函数，N为文本存在位置像素点的总数量，为所述目标掩膜，为所述超分辨率图片。

7.根据权利要求1所述的文本图像超分辨率重建方法，其特征在于，所述对所述文本掩膜进行上采样，获得目标掩膜的步骤包括：

对所述文本掩膜进行多倍上采样，获得所述目标掩膜。