[发明专利]图像超分方法

权利要求书

1.一种图像超分方法，其特征在于：包括以下步骤：

S101：将输入YUV图像的Y通道进行Y通道扩展；

S102：对所述步骤S101的输出进行超分推理；

S103：使用所述步骤S101的输出和所述步骤S102的输出进行残差计算；

S104：对所述步骤S103的输出进行像素清洗，输出超分Y通道图像；

S105：将所述输入YUV图像的UV通道进行UV通道扩展；

S106：将所述步骤S104的输出与所述步骤S105的输出合并后输出超分YUV图像。

2.如权利要求1所述的方法，其特征在于：

所述步骤S101-S104通过第一模型（100）执行，所述第一模型（100）设置在NPU、DSP或GPU上；

所述步骤S105-S106运行在DSP、CPU或GPU上。

3.如权利要求2所述的方法，其特征在于：

所述第一模型（100）包括：

运算层，包括：

卷积运算：进行卷积运算输出运算结果；

激活函数运算：对所述卷积运算的输出的所述运算结果，使用激活函数进行映射计算；

所述运算层为M层；M≥2；

残差计算：

将第1层所述运算层的输出与最后一层所述运算层的输出进行残差运算；

像素清洗：

对所述残差计算的输出进行像素清洗，输出超分Y通道图像。

4.如权利要求2所述的方法，其特征在于：

在存储空间上设置第一存储空间、第二存储空间；

所述第一存储空间储存所述输入YUV图像；所述CPU和所述GPU通过指针访问所述第一存储空间；

所述第二存储空间储存所述超分YUV图像；所述CPU和所述GPU通过指针访问所述第二存储空间。

5.如权利要求1所述的方法，其特征在于：

所述步骤S102-S103通过第二模型（200）执行，所述第二模型（200）设置在NPU、DSP或GPU上；

所述步骤S101、S104-S106运行在DSP、CPU或GPU上。

6.如权利要求5所述的方法，其特征在于：

所述第二模型（200）包括：

运算层，包括：

卷积运算：进行卷积运算输出运算结果；

激活函数运算：对所述卷积模块的输出结果，使用激活函数进行映射计算；

所述运算层为M层；M≥2；第1层所述运算层的输入为所述步骤S101的输出；

残差计算：

将所述步骤S101的输出和最后一层所述运算层的输出进行残差运算。

7.如权利要求5所述的方法，其特征在于：

所述步骤S101在CPU上执行时，执行方法为：对所述输入YUV图像的Y通道像素，依次进行处理；

所述步骤S101在GPU上执行时，执行方法为，对所述输入YUV图像的Y通道像素，启动k个线程执行指令集进行并行处理；k=像素点数/n。

8.如权利要求5所述的方法，其特征在于：

所述步骤S104在CPU上执行时，对所述步骤S103输出的像素，依次进行处理；

所述步骤S104在GPU上执行时，对所述步骤S103输出的像素，启动k个线程执行指令集进行并行处理；k=像素点数/n。

9.如权利要求2或5所述的方法，其特征在于：

所述步骤S105在CPU上执行，执行方法为，对所述输入YUV图像的UV通道的像素，依次进行处理；

所述步骤S105在GPU上执行，执行方法为，对所述输入YUV图像的UV通道的像素，启动k个线程执行指令集进行并行处理；k=像素点数/n。

10.如权利要求5所述的方法，其特征在于：

在存储空间上设置第一存储空间、第二存储空间、第三存储空间；

所述第一存储空间储存所述输入YUV图像；所述CPU和所述GPU通过指针访问所述第一存储空间；

所述第二存储空间存储所述超分Y通道图像；所述CPU和所述GPU通过指针访问所述第二存储空间；

所述第三存储空间储存所述超分YUV图像；所述CPU和所述GPU通过指针访问所述第三存储空间。