[发明专利]建立图像增强模型与图像增强的方法、装置

权利要求书

1.一种建立图像增强模型的方法，包括：

获取训练数据，所述训练数据中包含多个视频帧以及对应各视频帧的标准图像；

构建由特征提取模块、至少一个通道膨胀卷积模块与空间上采样模块组成的神经网络模型，其中每个通道膨胀卷积模块包含空间下采样子模块、通道膨胀子模块与空间上采样子模块；

使用各视频帧以及对应各视频帧的标准图像对所述神经网络模型进行训练，直至所述神经网络模型收敛，得到图像增强模型。

2.根据权利要求1所述的方法，其中，所述构建由特征提取模块、至少一个通道膨胀卷积模块与空间上采样模块组成的神经网络模型包括：

构建包含第一逐通道卷积层与第一逐点卷积层的空间下采样子模块，所述第一逐通道卷积层与第一逐点卷积层的通道数为第一通道数。

3.根据权利要求1所述的方法，其中，所述构建由特征提取模块、至少一个通道膨胀卷积模块与空间上采样模块组成的神经网络模型包括：

构建包含第一通道膨胀层、第二通道膨胀层与通道收缩层的通道膨胀子模块；

所述第一通道膨胀层包含第二逐通道卷积层与第二逐点卷积层，所述第二逐通道卷积层与第二逐点卷积层的通道数为第二通道数；

所述第二通道膨胀层包含第三逐点卷积层，所述第三逐点卷积层的通道数为第三通道数；

所述通道收缩层包含第四逐通道卷积层与第四逐点卷积层，所述第四逐通道卷积层与第四逐点卷积层的通道数为第一通道数。

4.根据权利要求1所述的方法，其中，所述构建由特征提取模块、至少一个通道膨胀卷积模块与空间上采样模块组成的神经网络模型包括：

构建包含第五逐通道卷积层与第五逐点卷积层的空间上采样子模块，所述第五逐通道卷积层与第五逐点卷积层的通道数为第一通道数。

5.根据权利要求1所述的方法，其中，所述使用各视频帧以及对应各视频帧的标准图像对所述神经网络模型进行训练，直至所述神经网络模型收敛包括：

获取各视频帧对应的相邻视频帧；

将各视频帧与各视频帧对应的相邻视频帧作为所述神经网络模型的输入，得到所述神经网络模型针对各视频帧的输出结果；

根据各视频帧的输出结果与对应各视频帧的标准图像计算损失函数；

在确定计算得到的损失函数收敛的情况下，完成对所述神经网络模型的训练。

6.根据权利要求1所述的方法，还包括，

在使用各视频帧以及对应各视频帧的标准图像对所述神经网络模型进行训练，直至所述神经网络模型收敛之后，确定已收敛的神经网络模型是否满足预设的训练要求；

若是，则停止训练，得到图像增强模型；

否则，将预设个数的通道膨胀卷积模块添加至神经网络模型中通道膨胀卷积模块的末尾；

使用各视频帧以及对应各视频帧的标准图像对添加通道膨胀卷积模块之后的神经网络模型进行训练；

在确定该神经网络模型收敛之后，转至执行确定已收敛的神经网络模型是否满足预设的训练要求的步骤，以此循环进行，直至确定已收敛的神经网络模型满足预设的训练要求。

7.一种图像增强的方法，包括：

获取待处理视频帧；

将所述待处理视频帧作为图像增强模型的输入，将所述图像增强模型的输出结果作为所述待处理视频帧的图像增强结果；

其中，所述图像增强模型是根据权利要求1-6中任一项方法预先训练得到的。

8.根据权利要求7所述的方法，其中，所述将所述待处理视频帧作为图像增强模型的输入包括：

获取所述待处理视频帧的相邻视频帧；

将所述待处理视频帧与所述相邻视频帧作为所述图像增强模型的输入。