python实现ps的饱和度调整公式

公式参考：

1.https://blog.csdn.net/xingyanxiao/article/details/48035537

2. https://www.cnblogs.com/muyuge/p/6152396.html

上述两个分别实现了matlab和c版本的，其中

1. 在rgb通道上操作，对每个像素点操作，调整范围-1-1之间；

2. A. delta=rgbmax-rgbmin；判断delta为0，continue；B. value=（rgbmax+rgbmin）/255；L=value/2；（HSL）；

 

# -*- coding: utf-8 -*-
"""
Created on Mon Jul  8 10:17:34 2019

@author: wxx
"""
# bao he du(saturability)

import cv2
import numpy as np
from math import ceil

img = cv2.imread("data/00001.jpg")
img = img.astype(np.float)

r = img[:,:,2]
g = img[:,:,0]
b = img[:,:,1]

m, n, c = img.shape
r_new = r
g_new = g
b_new = b



img_new = np.zeros(img.shape)

increment = 50/100

for i in range(m):
    for j in range(n):
        rgbmax = max(r[i,j], max(g[i, j], b[i, j]))
        rgbmin = min(r[i,j], min(g[i, j], b[i, j]))
        delta = (rgbmax-rgbmin)/255
        if delta==0: continue
        value = (rgbmax+rgbmin)/255
        L = value/2
        
        if L<0.5: S = delta/value
        else: S=delta/(2-value)
        
        if increment>=0:
            if (increment+S)>=1:
                alpha = S
            else:
                alpha = 1-increment
            alpha =1/alpha-1
            r_new[i, j] = r[i, j] +(r[i,j]-L*255)*alpha
            g_new[i, j] = g[i, j] +(g[i,j]-L*255)*alpha
            b_new[i, j] = b[i, j] +(b[i,j]-L*255)*alpha
        else:
            alpha = increment
            r_new[i, j] = L*255 +(r[i,j]-L*255)*alpha
            g_new[i, j] = L*255 +(g[i,j]-L*255)*alpha
            b_new[i, j] = L*255 +(b[i,j]-L*255)*alpha
            
img_new[:,:,2]=r_new
img_new[:,:,1]=b_new
img_new[:,:,0]=g_new        

import matplotlib.pyplot as plt
plt.imshow(img_new[:,:,0])

cv2.imwrite("satuablity-50%.png", img_new)