OpenCV调整彩色图像的饱和度和亮度

问题

如何调整彩色图像的饱和度和亮度

解决思路

详细步骤：

1. 将RGB图像值归一化到[0, 1]
2. 然后使用函数cvtColor进行色彩空间的转换
3. 接下来可以根据处理灰度图像对比度增强伽马变换或者线性变换调整饱和度和亮度分量
4. 最后转换到RGB色彩空间

代码

1# !/usr/bin/env python
2# -*-encoding: utf-8-*-
3# author:LiYanwei
4# version:0.1
5
6
7import numpy as np
8import cv2
9
10
11def main():
12    # 加载图片 读取彩色图像
13    image = cv2.imread('./Files_image/img1.jpg', cv2.IMREAD_COLOR)
14    # print(image)
15    # cv2.imshow("image", image)
16    # 图像归一化，且转换为浮点型
17    fImg = image.astype(np.float32)
18    fImg = fImg / 255.0
19    # 颜色空间转换 BGR转为HLS
20    hlsImg = cv2.cvtColor(fImg, cv2.COLOR_BGR2HLS)
21    l = 100
22    s = 100
23    MAX_VALUE = 100
24    # 调节饱和度和亮度的窗口
25    cv2.namedWindow("l and s", cv2.WINDOW_AUTOSIZE)
26    def nothing(*arg):
27        pass
28    # 滑动块
29    cv2.createTrackbar("l", "l and s", l, MAX_VALUE, nothing)
30    cv2.createTrackbar("s", "l and s", s, MAX_VALUE, nothing)
31    # 调整饱和度和亮度后的效果
32    lsImg = np.zeros(image.shape, np.float32)
33    # 调整饱和度和亮度
34    while True:
35        # 复制
36        hlsCopy = np.copy(hlsImg)
37        # 得到 l 和 s 的值
38        l = cv2.getTrackbarPos('l', 'l and s')
39        s = cv2.getTrackbarPos('s', 'l and s')
40        # 1.调整亮度（线性变换) , 2.将hlsCopy[:, :, 1]和hlsCopy[:, :, 2]中大于1的全部截取
41        hlsCopy[:, :, 1] = (1.0 + l / float(MAX_VALUE)) * hlsCopy[:, :, 1]
42        hlsCopy[:, :, 1][hlsCopy[:, :, 1] > 1] = 1
43        # 饱和度
44        hlsCopy[:, :, 2] = (1.0 + s / float(MAX_VALUE)) * hlsCopy[:, :, 2]
45        hlsCopy[:, :, 2][hlsCopy[:, :, 2] > 1] = 1
46        # HLS2BGR
47        lsImg = cv2.cvtColor(hlsCopy, cv2.COLOR_HLS2BGR)
48        # 显示调整后的效果
49        cv2.imshow("l and s", lsImg)
50
51        ch = cv2.waitKey(5)
52        # 按 ESC 键退出
53        if ch == 27:
54            break
55        elif ch == ord('s'):
56            # 按 s 键保存并退出
57            # 保存结果
58            lsImg = lsImg * 255
59            lsImg = lsImg.astype(np.uint8)
60            cv2.imwrite("lsImg.jpg", lsImg)
61            break
62
63    # 关闭所有的窗口
64    cv2.destroyAllWindows()
65
66
67if __name__ == "__main__":
68    main()