[发明专利]图像增强方法、装置及电子设备

权利要求书

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

获取原始图像对应的待处理图像并提取所述原始图像的色彩信息以及饱和度信息；

将所述待处理图像输入至亮度增强模型中，得到增强后的亮度信息；

对所述色彩信息、所述饱和度信息以及所述增强后的亮度信息进行融合，确定增强后的所述待处理图像；

其中，所述将所述待处理图像输入至亮度增强模型中，得到增强后的亮度信息，包括：

利用所述亮度增强模型中的至少一个卷积单元对所述待处理图像进行至少一次的特征提取；

利用所述亮度增强模型中的至少一个注意力机制单元对所述卷积单元或转置卷积单元的输出信息进行图像细节的恢复；

利用所述亮度增强模型中的至少一个转置卷积单元对所述注意力机制单元的输出信息进行解码；

将与所述亮度增强模型的输入层连接的第一个卷积单元的输出信息与最后一个所述注意力机制单元的输出信息进行融合，确定所述增强后的亮度信息。

2.根据权利要求1所述的方法，其特征在于，所述利用所述亮度增强模型中的至少一个注意力机制单元对所述卷积单元或转置卷积单元的输出信息进行图像细节的恢复，包括：

利用所述注意力机制单元中的卷积层对所述卷积单元的输出信息，或所述转置卷积单元的输出信息进行卷积处理，得到第一输出信息；

对所述第一输出信息进行全局池化以及全连接处理，得到第二输出信息；

计算所述第一输出信息与所述第二输出信息的乘积，得到对应于所述卷积单元的输出信息的图像细节恢复结果或对应于所述转置卷积单元的输出信息的图像细节恢复结果。

3.根据权利要求1所述的方法，其特征在于，所述将所述待处理图像输入至亮度增强模型中，得到增强后的亮度信息，包括：

将所述待处理图像输入所述亮度增强模型的第一个卷积单元，得到第一个卷积单元的特征提取结果；

将所述第一个卷积单元的特征提取结果输入第一卷积单元中，得到第一卷积单元的特征提取结果，所述第一卷积单元包括至少一个卷积单元；

将第一卷积单元的特征提取结果以及第一转置卷积单元的解码结果输入所述第一注意力机制单元中，得到图像细节恢复的结果；其中，所述第一转置卷积单元的输入为第二注意力机制单元的输出；

将所述第一注意力机制单元的输出与所述第一个卷积单元的输出进行融合，确定增强后的亮度信息。

4.根据权利要求1所述的方法，其特征在于，所述将与所述亮度增强模型的输入层连接的第一个卷积单元的输出信息与最后一个所述注意力机制单元的输出信息进行融合，确定所述增强后的亮度信息，包括：

计算所述第一个卷积单元的输出信息与最后一个所述注意力机制单元的输出信息之和，得到第三输出信息；

对所述第三输出信息进行卷积处理，得到所述增强后的亮度信息。

5.根据权利要求1所述的方法，其特征在于，所述获取原始图像对应的待处理图像，包括：

获取原始图像；

对所述原始图像进行预处理得到所述待处理图像，所述待处理图像中的各个像素点按预设规律排列。

6.根据权利要求1-5中任一项所述的方法，其特征在于，所述亮度增强模型是通过如下方式训练得到的：

获取至少一个场景的两种亮度信息的样本图像，得到与第一亮度信息对应的第一样本图像，以及与第二亮度信息对应的第二样本图像；

将所述第一样本图像输入亮度增强模型中，得到增强后的亮度信息；

基于增强后的亮度信息与对应的所述第二样本图像的亮度信息，对所述亮度增强模型的参数进行更新。

7.根据权利要求6所述的方法，其特征在于，所述获取至少一个场景的两种亮度信息的样本图像，得到与第一亮度信息对应的第一样本图像，以及与第二亮度信息对应的第二样本图像，包括：

获取至少一个场景的两种亮度信息的原始样本图像；

对所述原始样本图像进行预处理，得到所述第一样本图像以及所述第二样本图像，所述第一样本图像以及所述第二样本图像中的各个像素点按预设规律排列。