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

权利要求书

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

电子设备获取彩色图像，所述彩色图像包括第一Y通道分量、第一U通道分量和第一V通道分量；

所述电子设备对所述第一U通道分量和所述第一V通道分量进行图像尺寸放大，得到处理后的第二U通道分量和第二V通道分量，并对所述第一Y通道分量进行超分辨率处理，得到处理后的第二Y通道分量；

所述电子设备对所述彩色图像进行高动态范围成像处理，得到分别与所述第一Y通道分量、所述第一U通道分量和所述第一V通道分量分别对应的分量系数；

所述电子设备根据所述第二Y通道分量和所述第一Y通道分量对应的分量系数，得到与所述第一Y通道分量对应的增强后通道分量，并所述第二U通道分量和所述第一U通道分量对应的分量系数，得到与所述第一U通道分量对应的增强后通道分量，并所述第二V通道分量和所述第一V通道分量对应的分量系数，得到与所述第一V通道分量对应的增强后通道分量；

所述电子设备根据所述第一Y通道分量对应的增强后通道分量、所述第一U通道分量对应的增强后通道分量和所述第一V通道分量对应的增强后通道分量，合成增强后的彩色图像。

2.根据权利要求1所述的方法，其特征在于，

所述第一Y通道分量对应的增强后通道分量满足：Y‘＝W_Y×Y；其中，Y‘表示所述第一Y通道分量对应的增强后通道分量，W_Y表示与所述第一Y通道分量对应的分量系数，Y表示所述第二Y通道分量；

所述第一U通道分量对应的增强后通道分量满足：其中，U‘表示所述第一U通道分量对应的增强后通道分量，W_U表示与所述第一U通道分量对应的分量系数，U表示所述第二U通道分量，为固定参数，为固定参数，和的取值范围为[0,1]；

所述第一V通道分量对应的增强后通道分量满足：其中，V‘表示所述第一V通道分量对应的增强后通道分量，W_V表示所述第一V通道分量对应的分量系数，V表示所述第二V通道分量,为固定参数，为固定参数，和的取值范围为[0,1]。

3.根据权利要求1或2所述的方法，其特征在于，所述第一U通道分量对应的分量系数和所述第一V通道对应的分量系数相同。

4.根据权利要求1至3任一项所述的方法，其特征在于，所述电子设备包括嵌入式神经网络处理器NPU和中央处理器CPU，所述NPU集成有双分支多任务神经网络模型；

所述电子设备的NPU对输入至所述双分支多任务神经网络模型中的所述第一Y通道分量进行超分辨率处理，输出所述第二Y通道分量，并且所述电子设备的CPU对所述第一U通道分量和所述第一V通道分量进行图像尺寸放大，得到处理后的第二U通道分量和第二V通道分量；

所述电子设备的NPU对输入至所述双分支多任务神经网络模型中的所述彩色图像进行高动态范围成像处理，输出分别与所述第一Y通道分量、所述第一U通道分量和所述第一V通道分量分别对应的分量系数。

5.根据权利要求1至4任一项所述的方法，其特征在于，所述电子设备获取彩色图像，包括：

所述电子设备的视频解码器对输入的视频流进行解码，生成所述彩色图像，其中，所述输入的视频流中每帧图像为RGB颜色空间下的彩色图像。

6.根据权利要求1至4任一项所述的方法，其特征在于，所述电子设备合成增强后的彩色图像之后，还包括：

所述电子设备的视频编码器对各帧所述增强后的彩色图像进行编码，生成图像增强处理后的视频流，所述图像增强处理后的视频流中每帧图像为RGB颜色空间下的彩色图像。