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

权利要求书

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

获取待色彩增强的原始图像；

对所述原始图像执行前景检测操作，获得所述原始图像对应的前景置信度图像；

对所述原始图像执行色彩增强处理，获得所述原始图像对应的普通色彩增强图像；

对所述原始图像、所述前景置信度图像以及所述普通色彩增强图像执行融合处理，得到所述原始图像对应的显著色彩增强图像。

2.根据权利要求1所述的方法，其特征在于，对所述原始图像执行前景检测操作，获得所述原始图像对应的前景置信度图像，包括：

将所述原始图像输入预先训练的显著性检测网络；所述显著性检测网络用于检测所述原始图像的前景，输出所述原始图像对应的前景置信度图像；

获取所述显著性检测网络输出的所述原始图像对应的前景置信度图像。

3.根据权利要求1所述的方法，其特征在于，对所述原始图像执行色彩增强处理，获得所述原始图像对应的普通色彩增强图像，包括：

将所述原始图像输入预先训练的色彩增强网络；所述色彩增强网络用于对所述原始图像进行色彩增强处理，输出所述原始图像对应的普通色彩增强图像；

获取所述色彩增强网络输出的所述原始图像对应的普通色彩增强图像。

4.根据权利要求1所述的方法，其特征在于，对所述原始图像、所述前景置信度图像以及所述普通色彩增强图像执行融合处理，得到显著色彩增强图像，包括：

叠加所述原始图像、所述前景置信度图像以及所述普通色彩增强图像，得到叠加图像；

提取所述叠加图像中的前景背景特征和颜色分布特征；

根据所述前景背景特征、所述颜色分布特征以及所述原始图像的像素信息，生成所述显著色彩增强图像。

5.根据权利要求4所述的方法，其特征在于，根据所述前景背景特征、所述颜色分布特征以及所述原始图像的像素信息，生成所述显著色彩增强图像，包括：

根据所述前景背景特征和所述原始图像的像素信息，确定所述原始图像中属于前景区域的像素点和所述原始图像中属于背景区域的像素点；

根据所述颜色分布特征，确定所述原始图像中前景区域像素点的色调和所述原始图像中背景区域像素点的色调；

根据所述前景区域像素点的色调，调整所述前景区域像素点的色彩强度，根据所述背景区域像素点的色调，调整所述背景区域像素点的色彩强度，得到所述显著色彩增强图像；

其中，为所述前景区域像素点调整的色彩强度大于为所述背景区域像素点调整的色彩强度。

6.根据权利要求1所述的方法，其特征在于，对所述原始图像、所述前景置信度图像以及所述普通色彩增强图像执行融合处理，得到所述原始图像对应的显著色彩增强图像，包括：

将所述原始图像、所述前景置信度图像以及所述普通色彩增强图像以通道并联的方式输入预先训练的融合网络，获取所述融合网络输出的所述原始图像对应的显著色彩增强图像；其中，

所述融合网络用于对所述原始图像、所述前景置信度图像以及所述普通色彩增强图像执行融合处理。

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

所述融合网络包括：四个步幅为2的3×3卷积核、七个步幅为1的3×3卷积核、四个步幅为2的3×3反卷积核；其中，

所述融合网络从输入端到输出端顺序串联四个步幅为2的3×3卷积核、三个步幅为1的3×3卷积核、四个步幅为2的3×3反卷积核；

在剩余的四个步幅为1的3×3卷积核中，每个步幅为1的3×3卷积核的两端分别连接位置对应的一个步幅为2的3×3卷积核和一个步幅为2的3×3反卷积核。