饱和度调整算法说明(完整python代码在文末):

本算法主要是利用HSL颜色空间进行饱和度S的上下限控制,对RGB空间进行补丁式调整。调整过程在RGB空间进行,其原理简单地说就是判断每个像素的R、G、B值是否大于或小于平均值,大于加上调整值,小于则减去调整值,如何计算各个像素点的调整系数是关键。本算法主体思想就是利用HSL来计算各点的调整系数。

算法过程如下:

1. 首先将RGB空间图像转换为HSL空间,用来对饱和度S进行上下限控制,这里不考虑色相H差异。

转换公式如下:

            

python 将RGB保存为YUV python处理rgb_颜色空间

           

python 将RGB保存为YUV python处理rgb_python 将RGB保存为YUV_02

其中L表示亮度,S表示饱和度,min和max代表RGB空间中的R、G、B颜色值中的最小最大值,范围均为0-1的实数。

          

python 将RGB保存为YUV python处理rgb_python 将RGB保存为YUV_03

设置饱和度增量increment,范围为-1至1,并由此计算出适用于RGB空间的调整系数进行饱和度调整。

 

这里分两种情况:

1) 当增量i (increment) >= 0时,此时饱和度采用指数型增长方式调整。考虑到当前饱和度s与增量i之和可能大于1,超出饱和度上限,故仅仅考虑i作为调整依据不妥。在此选定当i+s大于1时,取s补数作为调整依据(s补数越大,调整越大);当i+s小于1,取i作为调整依据(i越大,调整越大)。RGB空间的调整系数alpha公式如下:


python 将RGB保存为YUV python处理rgb_颜色空间_04

    其中s代表当前饱和度,i代表设置的饱和度增量(increment)。

          

python 将RGB保存为YUV python处理rgb_python 将RGB保存为YUV_05

2) 当增量i (increment) < 0时,此时饱和度采用线性减弱方式调整。考虑到当i取-1时,R、G、B值会相等,可以直接采用线性的方式作调整。RGB空间的调整系数alpha如下:

            

python 将RGB保存为YUV python处理rgb_颜色空间_06

         

python 将RGB保存为YUV python处理rgb_颜色空间_07

 

3. 将饱和度调整过的RGB进行校验,即小于0的取0,大于1的取1。

          

python 将RGB保存为YUV python处理rgb_python 将RGB保存为YUV_08

4. 将饱和度调整完毕的RGB图输出比较即可。

 

完整python代码如下:

运行方式:打开终端界面,在该py文件目录下,运行:python  该文件.py  图片路径  饱和等级(-1~1)

例如:python  Saturation.py  C:\Users\PDD\Desktop\pdd.jpg  0.5

import cv2
import sys
import numpy as np
import matplotlib.pyplot as plt

"""
  PS算法
  主要是利用HSL颜色空间进行饱和度S和亮度L的上下限控制,对RGB空间进行补丁式调整。
  参考CSDN博客:
"""

def PSAlgorithm(rgb_img, increment):
    img = rgb_img * 1.0
    img_min = img.min(axis=2)
    img_max = img.max(axis=2)
    img_out = img
    
    #获取HSL空间的饱和度和亮度
    delta = (img_max - img_min) / 255.0
    value = (img_max + img_min) / 255.0
    L = value/2.0
    
    # s = L<0.5 ? s1 : s2
    mask_1 = L < 0.5
    s1 = delta/(value)
    s2 = delta/(2 - value)
    s = s1 * mask_1 + s2 * (1 - mask_1)
    
    # 增量大于0,饱和度指数增强
    if increment >= 0 :
        # alpha = increment+s > 1 ? alpha_1 : alpha_2
        temp = increment + s
        mask_2 = temp >  1
        alpha_1 = s
        alpha_2 = s * 0 + 1 - increment
        alpha = alpha_1 * mask_2 + alpha_2 * (1 - mask_2)
        
        alpha = 1/alpha -1 
        img_out[:, :, 0] = img[:, :, 0] + (img[:, :, 0] - L * 255.0) * alpha
        img_out[:, :, 1] = img[:, :, 1] + (img[:, :, 1] - L * 255.0) * alpha
        img_out[:, :, 2] = img[:, :, 2] + (img[:, :, 2] - L * 255.0) * alpha
        
    # 增量小于0,饱和度线性衰减
    else:
        alpha = increment
        img_out[:, :, 0] = img[:, :, 0] + (img[:, :, 0] - L * 255.0) * alpha
        img_out[:, :, 1] = img[:, :, 1] + (img[:, :, 1] - L * 255.0) * alpha
        img_out[:, :, 2] = img[:, :, 2] + (img[:, :, 2] - L * 255.0) * alpha
    
    img_out = img_out/255.0
    
    # RGB颜色上下限处理(小于0取0,大于1取1)
    mask_3 = img_out  < 0 
    mask_4 = img_out  > 1
    img_out = img_out * (1-mask_3)
    img_out = img_out * (1-mask_4) + mask_4
    
    return img_out

path = './resource/fruit.bmp'
increment = 0.5  # 范围-1到1

#  run : python Saturation.py (path) (increment)
if __name__ == "__main__":
    len = len(sys.argv)
    if len >= 2 :
        path = sys.argv[1]
        if len >= 3 :
            increment = float(sys.argv[2])

    img = cv2.imread(path)
    img = cv2.cvtColor(img,cv2.COLOR_BGR2RGB)
    img_new = PSAlgorithm(img, increment)

    plt.figure("img_original")
    plt.imshow(img/255.0)
    plt.axis('off')

    plt.figure("img_saturate")
    plt.imshow(img_new)
    plt.axis('off')

    plt.show()