[发明专利]模型训练方法、超分辨率感知方法及相关装置

权利要求书

1.一种模型训练方法，其特征在于，所述模型训练方法包括：

获取训练数据集，所述训练数据集中的每个训练数据包括一个低分辨率的训练图像和所述训练图像对应的高分辨率图像；

针对至少一个训练图像，对所述训练图像进行数据增广，得到所述训练图像对应的增广图像作为新的训练图像并存储至所述训练数据集；

基于所述训练图像对应的高分辨率图像，获取所述新的训练图像对应的高分辨率图像；

利用所述训练数据集，使用adam优化器训练预设深度神经网络，得到超分辨率感知模型，所述预设深度神经网络采用RCAN网络。

2.根据权利要求1所述的模型训练方法，其特征在于，所述基于所述训练图像对应的高分辨率图像，获取所述新的训练图像对应的高分辨率图像，包括：

获取所述训练图像对应的数据增广类型；

基于所述训练图像对应的高分辨率图像和数据增广类型，得到所述新的训练图像对应的高分辨率图像。

3.根据权利要求1所述的模型训练方法，其特征在于，所述预设深度神经网络包括浅特征提取模块、深度特征提取模块、上采样模块和重建模块；

所述利用所述训练数据集，使用adam优化器训练预设深度神经网络，得到超分辨率感知模型，包括：

将各所述训练图像输入所述浅特征提取模块，得到各所述训练图像对应的浅特征；

将各所述训练图像对应的浅特征输入所述深度特征提取模块，得到各所述训练图像对应的深度特征；

将各所述训练图像对应的深度特征输入所述上采样模块，得到各所述训练图像对应的上采样特征图；

将各所述训练图像对应的上采样特征图输入所述重建模块，得到各所述训练图像对应的重建图像；

基于各所述训练图像对应的重建图像和各所述训练图像对应的高分辨率图像，使用adam优化器对所述预设深度神经网络进行训练，得到超分辨率感知模型。

4.根据权利要求3所述的模型训练方法，其特征在于，所述深度特征提取模块包括G个残差组，G是大于1的整数；

所述将各所述训练图像对应的浅特征输入所述深度特征提取模块，得到各所述训练图像对应的深度特征，包括：

将各所述训练图像对应的第i-1深度中间信息输入第i个残差组，得到各所述训练图像对应的第i深度中间信息，以各所述训练图像对应的浅特征作为各所述训练图像对应的第0深度中间信息，i的取值分别是不大于G的每个正整数；

将各所述训练图像对应的第G深度中间信息输入第二卷积层，得到各所述训练图像对应的深度卷积中间信息；

利用各所述训练图像对应的浅特征和各所述训练图像对应的深度卷积中间信息，获取各所述训练图像对应的深度特征。

5.根据权利要求4所述的模型训练方法，其特征在于，每个残差组均包括B个残差块，B是大于1的整数；

所述将各所述训练图像对应的第i-1深度中间信息输入第i个残差组，得到各所述训练图像对应的第i深度中间信息，包括：

将各所述训练图像对应的第i个残差组的第j-1残差注意信息输入第i个残差组的第j个残差块，得到各所述训练图像对应的第i个残差组的第j残差注意信息，以各所述训练图像对应的第i-1深度中间信息作为各所述训练图像对应的第i个残差组的第0残差注意信息，j的取值分别是不大于B的每个正整数；

将各所述训练图像对应的第B残差注意信息输入第三卷积层，得到各所述训练图像对应的注意卷积中间信息；

利用各所述训练图像对应的第i-1深度中间信息和各所述训练图像对应的注意卷积中间信息，获取各所述训练图像对应的第i深度中间信息。