Python-Opencv中的直方图均衡化

Python-Opencv中的直方图均衡化

直方图均衡化

图像的直方图是对图像对比度效果上的一种处理,旨在使得图像整体效果均匀,黑与白之间的各个像素级之间的点更均匀一点。
直方图均衡化:如果一副图像的像素占有很多的灰度级而且分布均匀,那么这样的图像往往有高对比度和多变的灰度色调。直方图均衡化就是一种能仅靠输入图像直方图信息自动达到这种效果的变换函数。它的基本思想是对图像中像素个数多的灰度级进行展宽,而对图像中像素个数少的灰度进行压缩,从而扩展像元取值的动态范围,提高了对比度和灰度色调的变化,使图像更加清晰。
直方图均衡化只对灰度图进行处理。

直方图均衡化的两种方式

1. 全局直方图均衡化

全局直方图均衡化可能得到是一种全局意义上的均衡化,通过全局来增强对比度,提高图像质量。

函数
equalizeHist(src[, dst])

2. 局部直方图均衡化

但是有的时候全局直方图均衡化并不是很好,会把某些不该调整的部分给调整了。Opencv中还有一种直方图均衡化,它是一种局部直方图均衡化,也就是是说把整个图像分成许多小块(比如按10*10作为一个小块),那么对每个小块进行均衡化。

函数

createCLAHE([, clipLimit[, tileGridSize]])

其中:

  • clipLimit - 表示对比度的大小。

  • tileGridSize - 表示每次处理块的大小 。

两种直方图均衡化代码实现

import cv2 as cv
from matplotlib import pyplot as plt


def equalHist_demo(image):
	# 将图片变成灰度图
    gray = cv.cvtColor(image, cv.COLOR_BGR2GRAY)
    cv.imshow("gray", gray)
    # 全局直方图均衡函数
    dst = cv.equalizeHist(gray)
    plt.subplot(1, 4, 3)
    plt.xlim([0, 256])
    plt.ylim([0, 4000])
    plt.grid(color='r', linestyle='--', linewidth=0.5, alpha=0.3)
    plt.hist(gray.ravel(), 256, [0, 256])
    plt.title("equalHist_demo")
    cv.imshow("equalHist_demo", dst)

def clahe_demo(image):
    gray = cv.cvtColor(image, cv.COLOR_BGR2GRAY)
    plt.subplot(1,4,2)
    plt.hist(gray.ravel(), 256, [0,256])
    plt.title("gray")
    plt.xlim([0, 256])
    plt.ylim([0, 4000])
    plt.grid(color='r', linestyle='--', linewidth=0.5, alpha=0.3)
    # 局部直方图均衡化 对比度大小设置为5 处理块大小为8*8
    calhe = cv.createCLAHE(clipLimit=5.0, tileGridSize=(8,8))
    dst = calhe.apply(gray)
    plt.subplot(1, 4, 4)
    plt.grid(color='r', linestyle='--', linewidth=0.5, alpha=0.3)
    plt.hist(dst.ravel(), 256, [0, 256])
    plt.title("clahe_demo")
    plt.xlim([0, 256])
    plt.ylim([0, 4000])
    cv.imshow("clahe_demo", dst)


src = cv.imread("scenery.jpg")
cv.namedWindow("input image", cv.WINDOW_AUTOSIZE)
cv.imshow("input image", src)
# 创建画图区域 figure(num=None, figsize=None, dpi=None, facecolor=None, edgecolor=None, frameon=True)
# num:图像编号或名称,数字为编号 ,字符串为名称
# figsize:figsize:指定figure的宽和高,单位为英寸
# dpi:指定绘图对象的分辨率,即每英寸多少个像素,缺省值为80,1英寸等于2.5cm,A4纸是 21*30cm的纸张    
# facecolor:背景颜色     edgecolor:边框颜色      frameon:是否显示边框
plt.figure("scenery",figsize=(12, 5))
# .subplot()创建子图
plt.subplot(1,4,1)
# 给子图加上网格
plt.grid(color='r', linestyle='--', linewidth=0.5, alpha=0.3)
# 绘制直方图
plt.hist(src.ravel(), 256, [0, 256])
plt.title("original picture")
plt.xlim([0,256])
plt.ylim([0,4000])
equalHist_demo(src)
clahe_demo(src)
plt.show()
cv.waitKey(0)
cv.destroyAllWindows()

Python-Opencv中的直方图均衡化_第1张图片
(1,1)图为原图,(1,2)图为原图的灰度图,(2,1)图为经过全局直方图均衡化后的效果,(2,2)图为进过局部直方图均衡化后的效果。
Python-Opencv中的直方图均衡化_第2张图片
以上四个图分别为原图,灰度图,经全局直方均衡化,局部直方均衡化后的直方图。

"直接"对BGR图进行局部直方图均衡化

所谓“直接”,其实也是对每个通道的灰度图进行直方图均衡化,然后对均衡化的三个通道进行叠加即变成了BGR图。

import cv2 as cv
from matplotlib import pyplot as plt


# 对于plt.title(“中文”),不能再图一些标题显示中文,则在程序前加上以下两行代码即可:
plt.rcParams['font.sans-serif']=['SimHei']
plt.rcParams['axes.unicode_minus'] = False
image = cv.imread("sine.jpg")
cv.imshow("original picture", image)
plt.figure("图")
plt.subplot(2,2,1)
plt.title("原图直方图")
plt.hist(image.ravel(), 256, [0, 256])
color = ["b", "g", "r"]
plt.subplot(2,2,3)
plt.title("原图三通道直方图")
for i, color in enumerate(color):
    hist = cv.calcHist([image], [i], None, [256], [0, 256])
    plt.plot(hist, color = color)
# 分离每一个通道
b, g, r = cv.split(image)
# 创建局部直方图均衡化
clahe = cv.createCLAHE(clipLimit=2.0, tileGridSize=(5, 5))
# 对每一个通道进行局部直方图均衡化
b = clahe.apply(b)
g = clahe.apply(g)
r = clahe.apply(r)
# 合并处理后的三通道 成为处理后的图
image = cv.merge([b, g, r])
color = ["b", "g", "r"]
plt.subplot(2,2,4)
plt.title("三通道直方图均衡化")
for i, color in enumerate(color):
    hist = cv.calcHist([image], [i], None, [256], [0, 256])
    plt.plot(hist, color = color)
plt.subplot(2,2,2)
plt.title("直方图均衡化")
plt.hist(image.ravel(), 256, [0, 256])
cv.imshow("clahe", image)
plt.show()
cv.waitKey(0)
cv.destroyAllWindows()

Python-Opencv中的直方图均衡化_第3张图片
Python-Opencv中的直方图均衡化_第4张图片

注: 对于plt.title(“中文”),不能再图一些标题显示中文,则在程序前加上以下两行代码即可:
plt.rcParams[‘font.sans-serif’]=[‘SimHei’]
plt.rcParams[‘axes.unicode_minus’] = False

初学Opencv,如有错误地方和改进地方,真诚地邀请您提出来,谢谢!

本文结束…

你可能感兴趣的:(Python-Opencv中的直方图均衡化)