[发明专利]低光照图像增强方法、相关设备及可读存储介质

权利要求书

1.一种低光照图像增强方法，其特征在于，所述方法包括：

获取待进行增强的低光照图像；

将所述低光照图像输入分解模型，所述分解模型对所述低光照图像进行分解得到光照图；所述分解模型是基于训练用图像对训练得到的，所述训练用图像对包括一个训练用低光照图像，以及一个训练用正常光照图像，所述训练用低光照图像与所述训练用正常光照图像为同一场景下不完全配对的两个图像；

基于所述光照图，得到增强后的图像。

2.根据权利要求1所述的方法，其特征在于，所述分解模型的训练损失包括：

所述分解模型输出的所述训练用低光照图像的反射图的特征图与经过直方图均衡化处理的所述训练用低光照图像的特征图的一致性损失。

3.根据权利要求2所述的方法，其特征在于，所述训练损失还包括：

基于所述分解模型输出的所述训练用低光照图像的光照图得到的增强后的图像与所述训练用正常光照图像的判别器损失、所述分解模型输出的所述训练用低光照图像的反射图与所述分解模型输出的所述训练用正常光照图像的反射图的一致性损失、所述分解模型输出的所述训练用低光照图像的光照图和所述分解模型输出的所述训练用低光照图像的反射图的重建损失中的任意一种或多种。

4.根据权利要求1所述的方法，其特征在于，所述基于所述光照图，得到增强后的图像，包括：

基于所述光照图，确定所述低光照图像的曝光图；

确定所述低光照图像中照度良好的区域图，以及所述曝光图中照度良好的区域图；

将所述低光照图像中照度良好的区域图，以及所述曝光图中照度良好的区域图进行重构，得到增强后的图像。

5.根据权利要求4所述的方法，其特征在于，所述基于所述光照图，确定所述低光照图像的曝光图，包括：

基于所述光照图，确定所述低光照图像中低光照的像素点；

对所述低光照图像中低光照的像素点进行亮度变换，得到所述低光照图像的曝光图。

6.根据权利要求5所述的方法，其特征在于，所述基于所述光照图，确定所述低光照图像中低光照的像素点，包括：

将所述光照图中像素值小于预设阈值的像素点对应的所述低光照图像中的像素点，确定为所述低光照图像中低光照的像素点。

7.根据权利要求4所述的方法，其特征在于，所述确定所述低光照图像中照度良好的区域图，以及所述曝光图中照度良好的区域图，包括：

确定所述低光照图像对应的第一权重矩阵和第二权重矩阵；

基于所述第一权重矩阵与所述低光照图像，确定所述低光照图像中照度良好的区域图；

基于所述第二权重矩阵与所述曝光图，确定所述曝光图中照度良好的区域图。

8.根据权利要求7所述的方法，其特征在于，所述确定所述低光照图像对应的第一权重矩阵和第二权重矩阵，包括：

以所述光照图中各像素点的像素值为底数，以预设数值为指数，计算所述光照图中各像素点的权重，所述光照图中各像素点的权重组合得到所述第一权重矩阵；

对所述第一权重矩阵进行归一化计算，得到所述第二权重矩阵。

9.一种低光照图像增强装置，其特征在于，所述装置包括：

获取单元，用于获取待进行增强的低光照图像；

分解单元，用于将所述低光照图像输入分解模型，所述分解模型对所述低光照图像进行分解得到光照图；所述分解模型是基于训练用图像对训练得到的，所述训练用图像对包括一个训练用低光照图像，以及一个训练用正常光照图像，所述训练用低光照图像与所述训练用正常光照图像为同一场景下不完全配对的两个图像；

增强单元，用于基于所述光照图，得到增强后的图像。

10.一种低光照图像增强设备，其特征在于，包括存储器和处理器；

所述存储器，用于存储程序；

所述处理器，用于执行所述程序，实现如权利要求1至8中任一项所述的低光照图像增强方法的各个步骤。

11.一种可读存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时，实现如权利要求1至8中任一项所述的低光照图像增强方法的各个步骤。