(Python数字图像处理)彩色图像处理---色调和彩色校正以及直方图均衡化

文章目录


-基于Python+OpenCV,实验环境:pycharm+anaconda,参考《数字图像处理》冈萨雷斯第四版(初学图像处理和Python,欢迎指出错误~)

一、色调和彩色校正

彩色图像可以看做不同彩色通道图像的叠加,每一个通道都可以同灰度图像处理一样进行操作。由于numpy的矩阵操作功能很强大,所以处理彩色图像也是比较方便的。

若采用伽马变换,图像较亮,应该选择γ>1,压缩高灰度级,图像较暗,应选择γ<1,扩展低灰度级,增强对比度。
若采用S函数(对比度拉伸变换函数),选择合适的参数(斜率),能得到较高的对比度,S函数表达式为:s=T(r)=1/(1+(M/r)^E)
s是输出灰度,r是输入灰度,M是控制的灰度中值,E控制函数斜率。形状如下:
(Python数字图像处理)彩色图像处理---色调和彩色校正以及直方图均衡化
代码如下:

# -*- coding:utf-8 -*-
"""
作者:YJH
日期:2021年11月05日
"""
import matplotlib.pyplot as plt
import cv2 as cv
import numpy as np
from 彩色空间转换 import hsi2rgb              # 从前面写的一个文件里导入自定义的两个函数
from 彩色空间转换 import rgb2hsi

# 显示汉字用
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False


# 定义坐标数字字体及大小
def label_def():
    plt.xticks(fontproperties='Times New Roman', size=8)
    plt.yticks(fontproperties='Times New Roman', size=8)
    # plt.axis('off')                                      # 关坐标,可选


if __name__ == '__main__':
    # 读取图片
    img_orig = cv.imread('top_ left_flower.tif', 1)               # 读取彩色图片
# ------------------------------------------------色调校正---------------------------------------------------------#
    # 伽马变换处理
    img_gama = np.power(img_orig.astype(np.float32), 1.5)         # 图像较亮,若采用幂率变换,γ>1,压缩高灰度级
    temp1 = img_gama - np.min(img_gama)
    img_gama = temp1/np.max(temp1)
    # 对比度拉伸变换函数
    med = np.median(img_orig.astype(np.float32))                    # 获取中值M
    img_temp = 1 / (1 + np.power((140/(img_orig+1e-6)), 4.5))       # 4.5为斜率,交互式选择(感觉med效果不如140)
    temp2 = img_temp - np.min(img_temp)                             # 标定到[0~255],才能进行BGR2RGB
    img_con_str = np.uint8(255*(temp2/np.max(temp2)))
    # 显示所用的变换函数
    x1 = np.linspace(img_orig.min(), img_orig.max(), num=200)
    y1 = np.power(x1, 1.5)                                       # 伽马函数

    x2 = np.linspace(img_orig.min(), img_orig.max(), num=200)
    y2 = 1 / (1 + np.power((med/(x2+1e-6)), 4.5))               # 对比度拉伸函数

    plt.subplot(231), plt.title('原图像'), plt.imshow(cv.cvtColor(img_orig, cv.COLOR_BGR2RGB)), plt.axis('off')
    plt.subplot(232), plt.title('伽马变换'), plt.imshow(cv.cvtColor(img_gama, cv.COLOR_BGR2RGB)), plt.axis('off')
    plt.subplot(233), plt.title('对比度拉伸'), plt.imshow(cv.cvtColor(img_con_str, cv.COLOR_BGR2RGB)), plt.axis('off')
    plt.subplot(235), plt.title('s=r**(1.5)'), plt.plot(x1, y1), plt.grid(), label_def()
    plt.subplot(236), plt.title('s=1/(1+(M/r)**4)'), plt.plot(x2, y2), plt.grid(), label_def()
    plt.show()

效果如下:
(Python数字图像处理)彩色图像处理---色调和彩色校正以及直方图均衡化

二、色调校正及彩色平衡

只经过色调校正并不总能得到满意的结果。常用的处理方法是:
(1)色调校正;(2)彩色平衡校正。
比如下图较暗,所以用γ<1(0.5)的伽马变换来扩展低灰度级。但变换后图像中(前方石头和杂草)偏红色,所以转到CMY空间,对M分量进行平衡。代码如下:(用到的rgb2hsi和hsi2rgb是我自己定义的两个彩色空间变换函数,可以见我另一篇文章彩色空间HSI和RGB变换

# 接上面的代码
# --------------------------------------------彩色平衡---------------------------------------------------------------#
    img_stone = cv.imread('bottom_left_stream.tif', 1)
    # 伽马变换处理
    stone_gama = np.power(img_stone.astype(np.float32), 0.5)         # 图像较暗,若采用幂率变换,γ<1,拉伸低灰度级,交互式选择
    temp = stone_gama - np.min(stone_gama)
    stone_gama = temp/np.max(temp)

    img_cmy = 1 - cv.cvtColor(stone_gama, cv.COLOR_BGR2RGB)
    c, m, y = cv.split(img_cmy)
    # print(m.shape)
    m_gama = np.power(m.astype(np.float32), 1.08)                   # 深红色较多,压缩一下
    temp_m = m_gama - np.min(m_gama)
    m_gama = (temp_m/(np.max(temp_m)))
    out_stone = 1 - cv.merge((c, m_gama, y))

    plt.subplot(131), plt.title('原图像'), plt.imshow(cv.cvtColor(img_stone, cv.COLOR_BGR2RGB)), plt.axis('off')
    plt.subplot(132), plt.title('伽马变换'), plt.imshow(cv.cvtColor(stone_gama, cv.COLOR_BGR2RGB)), plt.axis('off')
    plt.subplot(133), plt.title('彩色平衡(深红色)'), plt.imshow(out_stone), plt.axis('off')
    plt.show()

效果如下(好像不明显还行吧
(Python数字图像处理)彩色图像处理---色调和彩色校正以及直方图均衡化

三、彩色直方图均衡化

同灰度图一样,可以直接用函数cv2.equalizeHist(img),操作RGB每个平面或者HSI空间的I分量。但是这个函数操作对象灰度级要是8bit的,对于[0,1]的灰度级要标定到[0,255],要注意一下。
HSI空间中I分量直方图均衡化后,虽不改变H和S分量,但会影响图像整体颜色。常用处理是先均衡化,再调整饱和度分量S。
代码如下:

# ------------------------彩色直方图均衡化----------------------------#
    img_caster = cv.imread('caster_stand_original.tif', 1)
    h, s, i, caster = rgb2hsi(img_caster)
    img = np.float32(caster)
    i = np.uint8(255*i)
    equ_i = (cv.equalizeHist(i))/255.0                                        # 均衡化亮度分量
    # plt.subplot(121), plt.imshow(i, 'gray')
    # plt.subplot(122), plt.imshow(equ_i, 'gray')
    # plt.show()
    img_equ1 = hsi2rgb(cv.merge((h, s, equ_i)))
    # add_s = np.where((s*2) > 1, s, (s*1.5))                                # 增饱和度
    add_s = np.power(s, 0.85)
    img_equ2 = hsi2rgb(cv.merge((h, add_s, equ_i)))

    plt.subplot(131), plt.title('原图像'), plt.imshow(cv.cvtColor(img_caster, cv.COLOR_BGR2RGB)), plt.axis('off')
    plt.subplot(132), plt.title('I分量直方图均衡'), plt.imshow(img_equ1), plt.axis('off')
    plt.subplot(133), plt.title('均衡I+增大S'), plt.imshow(img_equ2), plt.axis('off')
    plt.show()

效果如下(找不同…):
(Python数字图像处理)彩色图像处理---色调和彩色校正以及直方图均衡化
欢迎大家批评指正

上一篇:构建机器学习模型时要避免的6个错误


下一篇:【ANDROID游戏开发十九】(必看篇)SURFACEVIEW运行机制详解—剖析BACK与HOME按键及切入后台等异常处理!