openCV编程基础10--直方图应用

主要功能：

1.直方图均衡化(调整图像对比度)/(对灰度图而言的)/全局/局部的直方图均衡化，是图像对比度增强的一个手段

2.直方图比较,计算两张图片直方图的相关性，可以用于判断两张图片是否是很相似的

import cv2 as cv
import numpy as np


# 全局的直方图均衡化(对灰度图而言的)，是图像对比度增强的一个手段
def equalHist_demo(image):
    gray = cv.cvtColor(image,cv.COLOR_BGR2GRAY)
    det = cv.equalizeHist(gray)
    cv.imshow('gray',gray)
    cv.imshow('equalHist_demo',det)


# 局部自适应直方图均衡化(对灰度图而言的)，是图像对比度增强的一个手段
def clahe_demo(image):
    gray = cv.cvtColor(image,cv.COLOR_BGR2GRAY)
    clahe = cv.createCLAHE(clipLimit=2.0, tileGridSize=(8,8))
    det = clahe.apply(gray)
    cv.imshow('clahe_demo',det)


#创建彩色的三通道直方图
def creat_rgb_hist(image):
    h, w, c = image.shape
    rgbHist = np.zeros([16*16*16,1],np.float32)
    bsize = 256 / 16
    for row in range(h):
        for col in range(w):
            b = image[row, col, 0]
            g = image[row, col, 1]
            r = image[row, col, 2]
            index = np.int(b/bsize)*16*16 + np.int(g/bsize)*16 + np.int(r/bsize)
            rgbHist[np.int(index),0] = rgbHist[np.int(index),0]+1
    return rgbHist


#计算两张图片直方图的相关性（式距离/相关性/卡方）
def hist_compare(image1, image2):
    hist1 = creat_rgb_hist(image1)
    hist2 = creat_rgb_hist(image2)
    match1 = cv.compareHist(hist1, hist2, cv.HISTCMP_BHATTACHARYYA)
    match2 = cv.compareHist(hist1, hist2, cv.HISTCMP_CORREL)
    match3 = cv.compareHist(hist1, hist2, cv.HISTCMP_CHISQR)
    print('巴式距离：%s, 相关性：%s， 卡方：%s'%(match1, match2,match3))


src = cv.imread('F:001.jpg')
#cv.namedWindow('input_image', cv.WINDOW_AUTOSIZE)
cv.imshow("0", src)
equalHist_demo(src)      # 全局的直方图均衡化(对灰度图而言的)
clahe_demo(src)          # 局部自适应直方图均衡化
image2 = cv.imread('F:002.jpg')
hist_compare(src, image2)   #计算两张图片直方图相关性
cv.waitKey(0)
cv.destroyAllWindows()

输出结果：