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

权利要求书

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

获取原始图像；

生成所述原始图像对应的缩略图像；

将所述缩略图像输入至预先训练的颜色查找表生成模型，得到颜色查找表，所述颜色查找表是3D颜色查找表；

根据所述原始图像和所述颜色查找表确定所述原始图像对应的增强图像；

其中，所述颜色查找表生成模型通过如下步骤训练得到：

获取训练样本集合，其中，训练样本包括样本图像和对应的增强图像；

根据样本所包括的样本图像和对应的增强图像，生成用于将所述样本图像转化为所述增强图像的标注颜色查找表，将所述标注颜色查找表作为样本的标注；

确定初始化的颜色查找表生成模型，其中所述初始化的颜色查找表生成模型包括用于输出目标图像对应的颜色查找表的目标层；

利用机器学习的方法，将所述训练样本集合中的训练样本中的样本图像作为初始化的颜色查找表生成模型的输入，将与输入的样本图像对应的标注颜色查找表作为初始化的颜色查找表生成模型的期望输出，训练得到所述颜色查找表生成模型。

2.根据权利要求1所述的方法，其特征在于，根据所述原始图像和所述颜色查找表确定所述原始图像对应的增强图像包括：分别将所述原始图像中各像素值输入所述颜色查找表，将所述颜色查找表的输出结果作为所述增强图像对应像素的像素值。

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

获取摄像头采集到的照片，将所述照片缓存到缓冲区中作为所述原始图像。

4.根据权利要求3所述的方法，其特征在于，所述原始图像为包括人脸的原始图像。

5.根据权利要求1所述的方法，其特征在于，所述颜色查找表生成模型为基于卷积神经网络技术的机器学习模型。

6.一种图像增强的装置，其特征在于，包括：

原始图像获取单元，用于获取原始图像；

缩略图像生成单元，用于生成所述原始图像对应的缩略图像；

颜色查找表获取单元，用于将所述缩略图像输入至预先训练的颜色查找表生成模型，得到颜色查找表，所述颜色查找表是3D颜色查找表；

增强图像确定单元，用于根据所述原始图像和所述颜色查找表确定所述原始图像对应的增强图像；

其中，颜色查找表生成模型的训练装置，包括：

样本获取模块，用于获取训练样本集合，其中，训练样本包括样本图像和对应的增强图像；

样本标注模块，用于根据样本所包括的样本图像和对应的增强图像，生成用于将所述样本图像转化为所述增强图像的标注颜色查找表，将所述标注颜色查找表作为样本的标注；

模型确定模块，用于确定初始化的颜色查找表生成模型，其中所述初始化的颜色查找表生成模型包括用于输出目标图像对应的颜色查找表的目标层；

模型训练模块，用于利用机器学习的装置，将所述训练样本集合中的训练样本中的样本图像作为初始化的颜色查找表生成模型的输入，将与输入的样本图像对应的标注颜色查找表作为初始化的颜色查找表生成模型的期望输出，训练得到所述颜色查找表生成模型。

7.一种电子设备，其特征在于，包括：

一个或多个处理器；

存储器，用于存储一个或多个程序；

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现如权利要求1-5中任一项所述方法的指令。

8.一种计算机可读存储介质，其上存储有计算机程序，其特征在于，该计算机程序被处理器执行时实现如权利要求1-5任一项所述方法的步骤。