[发明专利]图像重建方法及装置

权利要求书

1.一种图像重建方法，其特征在于，包括：

获取预设Y通道图像和预设高分辨率图像；所述预设高分辨率图像的尺寸是所述预设Y通道图像的尺寸的h倍；所述预设高分辨率图像的像素个数是所述预设Y通道输入图像的像素个数的h²倍；

获取全卷积网；

通过全卷积网，对预设Y通道图像进行高分辨率重建，得到放大h倍的高分辨率Y通道图像；

获取高分辨率Y通道图像的特征和预设高分辨率图像的特征；

根据高分辨率Y通道图像的特征和预设高分辨率图像的特征，确定误差率；

当误差率小于或等于预设值时，将全卷积网作为预设全卷积网；当误差率大于预设值时，根据误差率更新所述全卷积网，直至确定的误差率小于或等于预设值；

获取图像中Y通道图像和UV通道图像；

通过预设全卷积网，对所述Y通道图像进行高分辨率重建，得到放大h倍的新Y通道图像；其中，所述新Y通道图像的像素个数是所述Y通道图像的像素个数的h²倍，所述h是大于1的数；

根据所述新Y通道图像和所述UV通道图像，生成放大所述h倍的新图像。

2.根据权利要求1所述的方法，其特征在于，所述根据所述新Y通道图像和所述UV通道图像，生成放大所述h倍的新图像包括：

通过预设插值方法，根据所述UV通道图像，得到放大所述h倍的新UV通道图像；

根据所述新Y通道图像和所述新UV通道图像，生成所述新图像。

3.根据权利要求1所述的方法，其特征在于，所述预设全卷积网包括n层卷积层，所述n为大于1的正整数，所述通过预设全卷积网，对所述Y通道图像进行高分辨率重建，得到所述h倍的新Y通道图像包括：

根据所述n层中前n-1层卷积层，将所述Y通道图像缩小m倍；所述m为大于1的数；

根据所述n层中第n层卷积层，将缩小后的Y通道图像放大mh倍，得到所述新Y通道图。

4.根据权利要求3所述的方法，其特征在于，所述n为4时，所述根据所述n层中前n-1层卷积层中的卷积核，将所述Y通道图像缩小m倍包括：

根据所述第一层卷积层的卷积核和长度为1的步长，提取所述Y通道图像的特征；

根据所述第二层卷积层的卷积核和长度为m的步长，对所述Y通道图像的特征进行降采样和增强，得到所述Y通道图像的新特征；

根据所述第三层卷积层的卷积核和长度为1的步长，对所述Y通道图像的新特征进行映射，得到所述缩小后的Y通道图像的新特征。

5.根据权利要求4所述的方法，其特征在于，所述根据所述n层中第n层卷积层，将缩小后的Y通道图像放大mh倍，得到所述新Y通道图包括：

根据所述第四层反卷积的卷积核和长度为所述mh的步长，对所述缩小后的Y通道图像的新特征进行升采样，得到所述新Y通道图像的特征。

6.根据权利要求1所述的方法，其特征在于，当所述特征为像素和语义特征时，所述根据所述高分辨率Y通道图像的特征和所述预设高分辨率图像的特征，确定误差率包括：

根据误差公式，确定所述误差率；所述误差公式为：

其中，所述L是所述误差率，所述L1是像素误差，所述L2是特征重建误差；所述是所述Y通道输出图像的第i个像素；所述q_j是所述预设高分辨率图像的第i个特征，所述M是所述Y通道输出图像的像素个数；所述是所述Y通道输出图像的第i维语义特征；所述r_k是所述预设高分辨率图像的第i维语义特征，所述N是所述Y通道输出图像的语义特征维数。