(19)中华 人民共和国 国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202111679517.4 (22)申请日 2021.12.31 (71)申请人 OPPO广东移动通信有限公司 地址 523860 广东省东莞 市长安镇乌沙海 滨路18号 (72)发明人 朱家成　 (74)专利代理 机构 深圳市隆天联鼎知识产权代 理有限公司 4 4232 代理人 刘抗美 (51)Int.Cl. G06T 5/00(2006.01) G06T 15/50(2011.01) G06T 17/00(2006.01) G06K 9/62(2022.01) G06N 3/04(2006.01)G06N 3/08(2006.01) G06V 10/80(2022.01) G06V 10/82(2022.01) (54)发明名称 图像色彩增强方法、 装置、 存储介质与电子 设备 (57)摘要 本公开提供一种图像色彩增强方法、 装置、 存储介质与电子设备， 涉及图像与视频处理技术 领域。 该图像色彩增强方法包括： 获取待处理图 像； 通过预先训练的深度神经网络对 所述待处理 图像提取基于三维 网格的特征， 并根据所提取的 特征生成信息矩阵， 所述三维网格是对所述待处 理图像的空域与像素值域所形成的三维空间进 行划分而得到的； 利用所述信息矩阵对所述待处 理图像进行色彩增强处理， 得到所述待处理图像 对应的色彩增强图像。 本公开提升了图像色彩增 强的效果。 权利要求书3页 说明书16页 附图10页 CN 114359100 A 2022.04.15 CN 114359100 A 1.一种图像色彩增强方法， 其特 征在于， 包括： 获取待处 理图像； 通过预先训练 的深度神经网络对所述待处理图像提取基于三维网格的特征， 并根据 所 提取的特征生成信息矩阵， 所述三维网格是对 所述待处理图像的空域与像素值域所形成的 三维空间进行划分而得到的； 利用所述信 息矩阵对所述待处理图像进行色彩增强处理， 得到所述待处理图像对应的 色彩增强图像。 2.根据权利要求1所述的方法， 其特征在于， 所述深度神经网络包括基础特征提取子网 络、 局部特征卷提取子网络、 全局特征提取子网络、 输出子网络； 所述通过预先训练的深度 神经网络对所述待处理图像提取基于三维网格的特征， 并根据所提取的特征生成信息矩 阵， 包括： 通过所述基础特征提取子网络按照空域网格的尺寸对所述待处理图像进行下采样处 理， 得到基础特 征， 所述空域网格为所述 三维网格在所述空域上的二维投影； 通过所述局部特 征卷提取子网络对所述基础特 征提取所述空域网格内的局部特 征； 通过所述全局特 征提取子网络对所述基础特 征提取全局特 征； 通过所述输出子网络按照 值域分区的数量对所述局部特征与 所述全局特征进行处理， 得到所述信息矩阵， 所述 值域分区为所述 三维网格在所述像素值 域上的一维投影。 3.根据权利要求2所述的方法， 其特征在于， 所述输出子网络包括融合层与维度转换 层； 所述通过所述输出子网络按照值域分区的数量对所述局部特征与所述全局特征进 行处 理， 得到所述信息矩阵， 包括： 通过所述融合层将所述局部特 征与所述全局特 征融合为综合特 征； 通过所述维度转换层按照所述值域分区的数量对所述综合特征进行维度转换， 得到所 述信息矩阵。 4.根据权利要求1所述的方法， 其特征在于， 所述信 息矩阵包括每个所述三维网格对应 的基准信息矩阵； 所述利用所述信息矩阵对所述待处理图像进行色彩增强处理， 得到所述 待处理图像对应的色彩增强图像， 包括： 基于所述待处理图像对所述基准信 息矩阵进行插值， 得到所述待处理图像的每个像素 点对应的色彩增强信息矩阵； 根据所述待处理图像的每个像素点对应的色彩增强信 息矩阵， 分别对所述待处理图像 的每个像素点进行处 理， 得到所述色彩增强图像。 5.根据权利要求4所述的方法， 其特征在于， 所述基于所述待处理图像对所述基准信 息 矩阵进行插值， 得到所述待处 理图像的每 个像素点对应的色彩增强信息矩阵， 包括： 当所述待处 理图像为多通道图像时， 将所述待处 理图像转换为单通道的参 考值图像； 基于所述参考值图像对所述基准信 息矩阵进行插值， 得到所述待处理图像的每个像素 点对应的色彩增强信息矩阵。 6.根据权利要求4所述的方法， 其特征在于， 所述基准信 息矩阵对应于所述三维网格的 中心点； 所述基于所述待处理图像对所述基准信息矩阵进行插值， 得到所述待处理图像的 每个像素点对应的色彩增强信息矩阵， 包括： 根据所述待处理图像的每个像素点相对于一个或多个所述三维网格的中心点的偏移，权　利　要　求　书 1/3 页 2 CN 114359100 A 2对一个或多个所述基准信息矩阵进行插值， 得到所述待处理图像的每个像素点对应的色彩 增强信息矩阵。 7.根据权利要求4所述的方法， 其特征在于， 所述根据所述待处理图像的每个像素点对 应的色彩增强信息矩阵， 分别对所述待处理图像的每个像素点进行处理， 得到所述色彩增 强图像， 包括： 根据所述色彩增强信 息矩阵的维度对所述待处理图像添加新通道， 并将所述新通道设 置为预设数值； 分别将所述待处理图像的每个像素点的像素值向量与所述每个像素点对应的色彩增 强信息矩阵相乘， 得到所述色彩增强 图像； 所述每个像素点的像素值向量为所述每个像素 点的各通道的数值所 形成的向量。 8.根据权利要求1所述的方法， 其特征在于， 所述待处理图像为图像序列中的一帧图 像； 所述方法还 包括： 利用所述信息矩阵对所述图像序列中包含所述待处理图像的子序列进行色彩增强处 理。 9.根据权利要求1所述的方法， 其特 征在于， 所述方法还 包括： 将待处理样本图像输入待训练的所述深度神经网络， 以输出样本信息矩阵； 利用所述样本信 息矩阵对所述待处理样本图像进行处理， 得到所述待处理样本图像对 应的色彩增强样本图像； 基于所述待处理样本图像对应的标注图像与所述色彩增强样本图像的差别， 更新所述 深度神经网络的参数。 10.根据权利要求1所述的方法， 其特 征在于， 所述方法还 包括： 将待处理样本图像输入待训练 的所述深度神经网络， 利用所述深度神经网络输出的第 一样本信息矩阵对所述待处理样本图像进行色彩增强处理， 得到色彩增强样本图像， 通过 变换参数对所述色彩增强样本图像进行变换， 得到第一变换样本图像； 通过所述变换参数对所述待处理样本图像进行变换， 将变换后的待处理样本图像输入 所述深度神经网络， 并利用所述深度神经网络输出的第二样本信息矩阵对 所述变换后的待 处理样本图像进行色彩增强处 理， 得到第二变换样本图像； 基于所述第 一变换样本图像与所述第 二变换样本图像的差别， 更新所述深度神经网络 的参数。 11.一种图像色彩增强装置， 其特 征在于， 包括： 图像获取模块， 被 配置为获取待处 理图像； 信息矩阵生成模块， 被配置为通过预先训练 的深度神经网络对所述待处理图像提取基 于三维网格的特征， 并根据所提取 的特征生成信息矩阵， 所述三维网格是对所述待处理图 像的空域与像素值 域所形成的三维空间进行划分而得到的； 色彩增强处理模块， 被配置为利用所述信息矩阵对所述待处理图像进行色彩增强处 理， 得到所述待处 理图像对应的色彩增强图像。 12.一种计算机可读存储介质， 其上存储有计算机程序， 其特征在于， 所述计算机程序 被处理器执行时实现权利要求1至10任一项所述的方法。 13.一种电子设备， 其特 征在于， 包括：权　利　要　求　书 2/3 页 3 CN 114359100 A 3