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

说明书

本发明涉及图像处理技术领域，具体设备及图像增强方法、装置及电子设备，所述方法包括获取原始图像对应的待处理图像并提取原始图像的色彩信息以及饱和度信息；将所述待处理图像输入至亮度增强模型中，得到增强后的亮度信息；对所述色彩信息、所述饱和度信息以及所述增强后的亮度信息进行融合，确定增强后的所述待处理图像。将待处理图像作为亮度增强模型的输入，输出为增强后的亮度信息，且将增强后的亮度信息与原始图像的色彩信息以及饱和度信息进行融合，所得到的增强后的待处理图像能保持待处理图像的色彩信息以及饱和度信息不变，仅对亮度信息进行增强，从而有效解决了增强后图像的偏色问题。

技术领域

本发明涉及图像处理技术领域，具体涉及图像增强方法、装置及电子设备。

背景技术

图像在较低的光照环境下或者背光的条件下往往会存在亮度低、噪点多以及对比差等问题，从主观上会影响到人们对图片的感受，同时也会影响后续在图像像素级别上的相关任务，如夜晚视频监控下的车牌识别、背光条件下的人脸识别等。通过图像增强技术能够减少图像中的噪声，提高物体与背景之间的对比度，改善图像的视觉效果同时也是后续相关识别任务的基础。因此，低照图像增强方法的研究具有比较重要的现实意义和价值。

目前所采用的图像增强方法是获取到图像传感器采集到的图像，将图像输入至图像增强模型中，输出为对输入图像的整体增强。然而，这一处理方式中，由于在对图像增强的过程中是对图像的整体进行增强的，就会导致增强处理后的图像容易出现偏色问题。

发明内容

有鉴于此，本发明实施例提供了一种图像增强方法、装置及电子设备，以解决增强后图像的偏色问题。

根据第一方面，本发明实施例提供了一种图像增强方法，包括：

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

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

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

本发明实施例提供的图像增强方法，将待处理图像作为亮度增强模型的输入，输出为增强后的亮度信息，且将增强后的亮度信息与原始图像的色彩信息以及饱和度信息进行融合，所得到的增强后的待处理图像能保持待处理图像的色彩信息以及饱和度信息不变，仅对亮度信息进行增强，从而有效解决了增强后图像的偏色问题。进一步地，由于亮度增强模型是仅对亮度信息进行增强，能够减少内存的开销，使得该图像增强方法能够应用在终端设备上，对终端设备采集到的图像进行实时的图像增强。

结合第一方面，在第一方面第一实施方式中，所述将所述待处理图像输入至亮度增强模型中，得到增强后的亮度信息，包括：

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

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

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

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

本发明实施例提供的图像增强方法，在亮度增强模型中利用转置卷积单元实现对注意力机制单元的输出信息的解码，利用注意力机制单元实现图像细节的恢复，从而该亮度增强模型能够更加准确地实现对待处理图像的亮度信息进行增强。

结合第一方面第一实施方式，在第一方面第二实施方式中，所述利用所述亮度增强模型中的至少一个注意力机制单元对所述卷积单元或转置卷积单元的输出信息进行图像细节的恢复，包括：