[发明专利]一种图像增强方法及电子设备

说明书

本申请提供一种图像增强方法及电子设备，用以实现图像增强，解决现有技术中存在的图像质量算法不满足移动端处理视频实时要求的问题。该方法包括：电子设备获取YUV彩色空间下的彩色图像，并且电子设备的CPU对该彩色图像的第一U通道分量和第一V通道分量进行图像尺寸放大，得到处理后的第二U通道分量和第二V通道分量，以及电子设备的NPU对该彩色图像的第一Y通道分量进行超分辨率处理，得到处理后的第二Y通道分量；另外，电子设备的NPU对彩色图像进行高动态范围成像处理，得到分别与三个通道分量分别对应的分量系数，最终，电子设备的CPU根据三个通道对应的分量系数和处理后的通道分量，合成增强后的彩色图像。

技术领域

本申请涉及图像处理技术领域，尤其涉及一种图像增强方法及电子设备。

背景技术

随着智能手机的普及，当前的移动端视频用户数量迅速增加，用户对于画面质量的要求也一直在提高，因此需要能够对视频中的亮度、对比度等参数进行自适应调整，优化并增强图像的画质。

图像增强是通过采用一系列图像增强技术去改善图像的质量和视觉效果，突出图像中感兴趣的特征，获得图像中有价值的信息，从而将图像转换成一种更适合于人或者机器进行分析和处理的形式，使得处理后的图像对某些特定的应用有更好的效果。图像增强理论广泛应用于生物医学领域、工业生产领域、公共安全领域以及航空航天领域等。

目前相关技术中，一种图像增强方法是使用一种超分辨率(super resolution，SR)的技术，将输入图像升频转化为高分辨率的输出图像。另一种图像增强方法则是使用高动态范围成像(high dynamic range imaging，HDR)技术，将一个标准动态范围的输入图像中的每个像素的取值，扩展到高动态范围下。对于智能手机而言，智能手机侧的视频解码器的解码结果通常都是YUV颜色空间，而不是RGB颜色空间，而目前的图像增强方法都是利用RGB颜色空间的视频进行模型训练和推断生成的，并不适用于智能手机等移动平台使用。而且目前的图像增强方法基于深度学习和机器学习的超分辨率算法，计算量较大，不能满足移动端处理视频的实时要求，功耗也较大。

发明内容

本申请提供一种图像增强方法及电子设备，用以实现更好的图像增强效果，解决现有技术中存在的图像质量算法不满足移动端处理视频的实时要求的问题。

第一方面，本申请实施例提供了一种图像增强方法，包括：电子设备获取YUV彩色空间下的彩色图像，因电子设备对该彩色图像的第一U通道分量和第一V通道分量进行图像尺寸放大，得到处理后的第二U通道分量和第二V通道分量，并对该彩色图像的第一Y通道分量进行超分辨率处理，得到处理后的第二Y通道分量；另外，电子设备对该彩色图像进行高动态范围成像处理，得到分别与第一Y通道分量、第一U通道分量和第一V通道分量分别对应的分量系数。

进一步地，电子设备根据所述第二Y通道分量和所述第一Y通道分量对应的分量系数，得到与所述第一Y通道分量对应的增强后通道分量，并所述第二U通道分量和所述第一U通道分量对应的分量系数，得到与所述第一U通道分量对应的增强后通道分量，并所述第二V通道分量和所述第一V通道分量对应的分量系数，得到与所述第一V通道分量对应的增强后通道分量，最终电子设备根据第一Y通道分量对应的增强后通道分量、第一U通道分量对应的增强后通道分量和第一V通道分量对应的增强后通道分量，合成增强后的彩色图像。

本申请实施例中，电子设备针对YUV颜色空间下的彩色图像的不同通道，采取不同的图像增强方法，既对YUV颜色空间下的彩色图像进行超分辨率处理，也进行高动态范围成像处理，提升了图像增强效果。

在一种可能的设计中，电子设备的CPU对该彩色图像的第一U通道分量和第一V通道分量进行放大，得到放大后的第二U通道分量和第二V通道分量，同时，电子设备的NPU对该彩色图像的第一Y通道分量进行超分辨率处理，得到处理后的第二Y通道分量；且电子设备的NPU对该彩色图像进行高动态范围成像处理，得到分别与三种通道对应的分量系数。可见，电子设备的CPU和NPU并行执行，可以提高图像的处理速度，以满足移动端处理视频的实时要求。