[发明专利]一种图像的色彩增强方法、装置、设备和存储介质

说明书

本发明公开了一种图像的色彩增强方法、装置、设备和存储介质。该方法包括：获取待色彩增强的原始图像；对所述原始图像执行前景检测操作，获得所述原始图像对应的前景置信度图像；对所述原始图像执行色彩增强处理，获得所述原始图像对应的普通色彩增强图像；对所述原始图像、所述前景置信度图像以及所述普通色彩增强图像执行融合处理，得到所述原始图像对应的显著色彩增强图像。本发明将前景置信度图像应用到色彩增加中，使得增强后的图像色彩更加鲜明，层次感更加分明。

技术领域

本发明涉及图像处理技术领域，尤其涉及一种图像的色彩增强方法、装置、设备和存储介质。

背景技术

随着人们观赏水平的不断提升，人们对图像画质的要求也不断提高，不但要求图像画质清晰，还要求图像的色彩鲜艳。由此，色彩增强技术随之产生，色彩增强技术可以对图像进行色彩调整，增加图像的鲜艳度。

目前，可以利用基于深度学习的端到端网络对图像进行色彩增强，得到该图像对应的色彩增强图像。进一步地，可以利用大量数据训练该端到端网络，使该端到端网络对图像进行色彩增强处理，生成图像对应的色彩增强图像。但是，在实际应用中发现，该端到端网络是对图像进行全图色彩增强，并且对全图做相同强度的色彩增强。这就容易导致原图中的背景或者次要景物被增强后反而比主要景物(前景)更鲜艳，从而改变了原图所突出的重点，使得色彩增强图像的层次感差，主次景物不分明。

发明内容

本发明的主要目的在于提供一种图像的色彩增强方法、装置、设备和存储介质，以解决利用现有色彩增强方式获得的色彩增强图像层次感差的问题。

针对上述技术问题，本发明是通过以下技术方案来解决的：

本发明提供了一种图像的色彩增强方法，包括：获取待色彩增强的原始图像；对所述原始图像执行前景检测操作，获得所述原始图像对应的前景置信度图像；对所述原始图像执行色彩增强处理，获得所述原始图像对应的普通色彩增强图像；对所述原始图像、所述前景置信度图像以及所述普通色彩增强图像执行融合处理，得到所述原始图像对应的显著色彩增强图像。

其中，对所述原始图像执行前景检测操作，获得所述原始图像对应的前景置信度图像，包括：将所述原始图像输入预先训练的显著性检测网络；所述显著性检测网络用于检测所述原始图像的前景，输出所述原始图像对应的前景置信度图像；获取所述显著性检测网络输出的所述原始图像对应的前景置信度图像。

其中，对所述原始图像执行色彩增强处理，获得所述原始图像对应的普通色彩增强图像，包括：将所述原始图像输入预先训练的色彩增强网络；所述色彩增强网络用于对所述原始图像进行色彩增强处理，输出所述原始图像对应的普通色彩增强图像；获取所述色彩增强网络输出的所述原始图像对应的普通色彩增强图像。

其中，对所述原始图像、所述前景置信度图像以及所述普通色彩增强图像执行融合处理，得到显著色彩增强图像，包括：叠加所述原始图像、所述前景置信度图像以及所述普通色彩增强图像，得到叠加图像；提取所述叠加图像中的前景背景特征和颜色分布特征；根据所述前景背景特征、所述颜色分布特征以及所述原始图像的像素信息，生成所述显著色彩增强图像。

其中，根据所述前景背景特征、所述颜色分布特征以及所述原始图像的像素信息，生成所述显著色彩增强图像，包括：根据所述前景背景特征和所述原始图像的像素信息，确定所述原始图像中属于前景区域的像素点和所述原始图像中属于背景区域的像素点；根据所述颜色分布特征，确定所述原始图像中前景区域像素点的色调和所述原始图像中背景区域像素点的色调；根据所述前景区域像素点的色调，调整所述前景区域像素点的色彩强度，根据所述背景区域像素点的色调，调整所述背景区域像素点的色彩强度，得到所述显著色彩增强图像；其中，为所述前景区域像素点调整的色彩强度大于为所述背景区域像素点调整的色彩强度。