[发明专利]超分辨图像生成方法及装置

权利要求书

1.一种超分辨图像生成方法，其特征在于，包括：

获取待处理的退化图像；

获取预设的神经网络模型，所述神经网络模型为采用训练数据对初始的神经网络模型进行训练后得到的；所述训练数据中包括：大于预设数量的原始图像，以及所述原始图像对应的退化图像；所述退化图像的退化类型包括以下类型中的任意一种或者多种：下采样类型、模糊类型；

采用预设的神经网络模型，对所述退化图像进行处理，得到与所述退化图像对应的超分辨图像。

2.根据权利要求1所述的方法，其特征在于，所述神经网络模型的训练方式为，

采用所述训练数据对初始的神经网络模型进行训练，得到训练后的神经网络模型；

对所述训练后的神经网络模型进行剪枝处理，对剪枝处理后的神经网络模型进行训练，直至满足预设条件；所述预设条件包括以下条件中的任意一个或者多个：所述神经网络模型训练的轮数大于预设轮数阈值、所述神经网络模型所占用的内存小于预设内存阈值、所述神经网络模型的处理速度大于预设速度阈值。

3.根据权利要求1所述的方法，其特征在于，所述训练数据中还包括：数据增强后的原始图像以及对应的数据增强后的退化图像。

4.根据权利要求1或3所述的方法，其特征在于，所述下采样类型设置有多个不同的缩放因子；

所述训练数据中原始图像对应的退化图像，为按照不同的缩放因子对原始图像进行下采样，和/或模糊处理，得到的退化图像。

5.根据权利要求1所述的方法，其特征在于，所述神经网络模型包括：多层神经网络；

所述获取预设的神经网络模型之后，还包括：

获取所述神经网络模型中各层神经网络变量的内存占用量；

按照所述内存占用量中的最大值，为所述神经网络模型分配内存；

按照分配的内存，对所述神经网络模型进行部署；

所述神经网络模型对待处理的退化图像进行处理的方式为，

将所述待处理的退化图像输入第一层神经网络，获取第一层神经网络输出的各个变量；

将第一层神经网络输出的各个变量缓存到分配的内存中；

将第一层神经网络输出的各个变量输入第二层神经网络，获取第二层神经网络输出的各个变量；

采用第二层神经网络输出的各个变量缓存到内存中，覆盖内存中缓存的各个变量，实现对内存的复用，直至获取到最后一层神经网络输出的超分辨图像。

6.一种超分辨图像生成装置，其特征在于，包括：

获取模块，获取待处理的退化图像；

所述获取模块，还用于获取预设的神经网络模型，所述神经网络模型为采用训练数据对初始的神经网络模型进行训练后得到的；所述训练数据中包括：大于预设数量的原始图像，以及所述原始图像对应的退化图像；所述退化图像的退化类型包括以下类型中的任意一种或者多种：下采样类型、模糊类型；

处理模块，用于采用预设的神经网络模型，对所述退化图像进行处理，得到与所述退化图像对应的超分辨图像。

7.根据权利要求6所述的装置，其特征在于，所述神经网络模型的训练方式为，

采用所述训练数据对初始的神经网络模型进行训练，得到训练后的神经网络模型；

对所述训练后的神经网络模型进行剪枝处理，对剪枝处理后的神经网络模型进行训练，直至满足预设条件；所述预设条件包括以下条件中的任意一个或者多个：所述神经网络模型训练的轮数大于预设轮数阈值、所述神经网络模型所占用的内存小于预设内存阈值、所述神经网络模型的处理速度大于预设速度阈值。

8.根据权利要求6所述的装置，其特征在于，所述训练数据中还包括：数据增强后的原始图像以及对应的数据增强后的退化图像。

9.根据权利要求6或8所述的装置，其特征在于，所述下采样类型设置有多个不同的缩放因子；

所述训练数据中原始图像对应的退化图像，为按照不同的缩放因子对原始图像进行下采样，和/或模糊处理，得到的退化图像。