目錄

• 概念
• • 直方圖定義
  • 直方圖均衡化
  • • 為什么要選用累積分布函數(shù)
    • 如何運(yùn)用累積分布函數(shù)使得直方圖均衡化
• C++ 源碼
• • 直方圖均衡化
  • 繪制直方圖
  • 主函數(shù)
• 效果
• 完整源碼

平臺(tái)：Windows 10 20H2
Visual Studio 2015
OpenCV 4.5.3

---

本文綜合自直方圖計(jì)算和程序員-圖哥——圖像處理之直方圖均衡化及C++實(shí)現(xiàn)

概念

直方圖定義

直方圖均衡化

直方圖均衡化的作用是圖像增強(qiáng)。

有兩個(gè)問(wèn)題比較難懂，一是為什么要選用累積分布函數(shù)，二是為什么使用累積分布函數(shù)處理后像素值會(huì)均勻分布。

為什么要選用累積分布函數(shù)

均衡化過(guò)程中，必須要保證兩個(gè)條件：
①像素?zé)o論怎么映射，一定要保證原來(lái)的大小關(guān)系不變，較亮的區(qū)域，依舊是較亮的，較暗依舊暗，只是對(duì)比度增大，絕對(duì)不能明暗顛倒；
②如果是八位圖像，那么像素映射函數(shù)的值域應(yīng)在0和255之間的，不能越界。
綜合以上兩個(gè)條件，累積分布函數(shù)是個(gè)好的選擇，因?yàn)槔鄯e分布函數(shù)是單調(diào)增函數(shù)（控制大小關(guān)系），并且值域是0到1（控制越界問(wèn)題），所以直方圖均衡化中使用的是累積分布函數(shù)。

如何運(yùn)用累積分布函數(shù)使得直方圖均衡化

直方圖均衡化過(guò)程中，映射方法是

其中，n是圖像中像素的總和，$n_k$是當(dāng)前灰度級(jí)的像素個(gè)數(shù)，L是圖像中可能的灰度級(jí)總數(shù)。

C++ 源碼

直方圖均衡化

直方圖均衡化的代碼實(shí)現(xiàn)有以下幾個(gè)步驟：

遍歷全圖，先統(tǒng)計(jì)每個(gè)灰度級(jí)下的像素點(diǎn)個(gè)數(shù)（為此我們開(kāi)辟了256大小的數(shù)組）；
計(jì)算每個(gè)灰度級(jí)的像素點(diǎn)占總像素的點(diǎn)的比例；
按照第二步求出的比例重新計(jì)算每個(gè)灰度級(jí)下的新的灰度值，即均衡化；
依照新的灰度值表遍歷更新圖像的灰度值。

Mat getEqualizeHistImage(Mat input) {int gray[256] = { 0 }; //記錄每個(gè)灰度級(jí)別下的像素個(gè)數(shù)double gray_prob[256] = { 0 }; //記錄灰度分布密度double gray_distribution[256] = { 0 }; //記錄累計(jì)密度int gray_equal[256] = { 0 }; //均衡化后的灰度值int gray_sum = 0; //像素總數(shù)Mat chRGB[3];split(input, chRGB);Mat output = input.clone();for (unsigned char k = 0; k < 3; ++k){for (unsigned short i = 0; i < 256; ++i){gray[i] = 0;gray_prob[i] = 0;gray_distribution[i] = 0;gray_equal[i] = 0;}gray_sum = chRGB[k].cols * chRGB[k].rows;//統(tǒng)計(jì)每個(gè)灰度下的像素個(gè)數(shù)for (int i = 0; i < chRGB[k].rows; i++){uchar* p = chRGB[k].ptr<uchar>(i);for (int j = 0; j < chRGB[k].cols; j++){int vaule = p[j];gray[vaule]++;}}//統(tǒng)計(jì)灰度頻率for (int i = 0; i < 256; i++){gray_prob[i] = ((double)gray[i] / gray_sum);}//計(jì)算累計(jì)密度gray_distribution[0] = gray_prob[0];for (int i = 1; i < 256; i++){gray_distribution[i] = gray_distribution[i - 1] + gray_prob[i];}//重新計(jì)算均衡化后的灰度值，四舍五入。參考公式：(N-1)*T+0.5for (int i = 0; i < 256; i++){gray_equal[i] = (uchar)(255 * gray_distribution[i] + 0.5);}//直方圖均衡化,更新原圖每個(gè)點(diǎn)的像素值for (int i = 0; i < chRGB[k].rows; i++){uchar* p = chRGB[k].ptr<uchar>(i);for (int j = 0; j < chRGB[k].cols; j++){p[j] = gray_equal[p[j]];}}}merge(chRGB, 3, output);return output; }

繪制直方圖

Mat getHistImage(Mat input) {Mat chRGB[3];split(input, chRGB);// 設(shè)定bin數(shù)目int histSize = 255;// 設(shè)定取值范圍 ( R,G,B) )float range[] = { 0, 255 };const float* histRange = { range };bool uniform = true; bool accumulate = false;Mat RGB_Hist[3];for(uint8_t i = 0; i < 3; ++i)// 計(jì)算直方圖:calcHist(&chRGB[i], 1, 0, Mat(), RGB_Hist[i], 1, &histSize, &histRange, uniform, accumulate);// 創(chuàng)建直方圖畫(huà)布int hist_w = 400; int hist_h = 400;int bin_w = cvRound((double)hist_w / histSize);Mat histImage(hist_w, hist_h, CV_8UC3, Scalar(0, 0, 0));for(uint8_t i = 0; i < 3; ++i)// 將直方圖歸一化到范圍 [ 0, histImage.rows ]normalize(RGB_Hist[i], RGB_Hist[i], 0, histImage.rows, NORM_MINMAX, -1, Mat());// 在直方圖畫(huà)布上畫(huà)出直方圖for (int i = 1; i < histSize; i++){line(histImage, Point(bin_w*(i - 1), hist_h - cvRound(RGB_Hist[0].at<float>(i - 1))),Point(bin_w*(i), hist_h - cvRound(RGB_Hist[0].at<float>(i))),Scalar(0, 0, 255), 2, 8, 0);line(histImage, Point(bin_w*(i - 1), hist_h - cvRound(RGB_Hist[1].at<float>(i - 1))),Point(bin_w*(i), hist_h - cvRound(RGB_Hist[1].at<float>(i))),Scalar(0, 255, 0), 2, 8, 0);line(histImage, Point(bin_w*(i - 1), hist_h - cvRound(RGB_Hist[2].at<float>(i - 1))),Point(bin_w*(i), hist_h - cvRound(RGB_Hist[2].at<float>(i))),Scalar(255, 0, 0), 2, 8, 0);}/// 顯示直方圖///imshow("calcHist Demo", histImage);return histImage; }

主函數(shù)

圖片路徑根據(jù)實(shí)際情況調(diào)整，注意反斜杠是轉(zhuǎn)義字符的開(kāi)頭，故“\”應(yīng)替換為“\\”

int main(int argc, char * argv[]) {Mat Image = imread("D:\\Work\\OpenCV\\Workplace\\Test_1\\1.jpg");imshow("原圖", Image);imshow("原圖直方圖", getHistImage(Image));imshow("直方圖均衡化后的圖像", getEqualizeHistImage(Image));imshow("直方圖均衡化后的直方圖", getHistImage(getEqualizeHistImage(Image)));waitKey(0);return 0; }

效果

完整源碼

#include <opencv2/opencv.hpp> #include <iostream>using namespace cv; using namespace std;Mat getHistImage(Mat input) {Mat chRGB[3];split(input, chRGB);// 設(shè)定bin數(shù)目int histSize = 255;// 設(shè)定取值范圍 ( R,G,B) )float range[] = { 0, 255 };const float* histRange = { range };bool uniform = true; bool accumulate = false;Mat RGB_Hist[3];for(uint8_t i = 0; i < 3; ++i)// 計(jì)算直方圖:calcHist(&chRGB[i], 1, 0, Mat(), RGB_Hist[i], 1, &histSize, &histRange, uniform, accumulate);// 創(chuàng)建直方圖畫(huà)布int hist_w = 400; int hist_h = 400;int bin_w = cvRound((double)hist_w / histSize);Mat histImage(hist_w, hist_h, CV_8UC3, Scalar(0, 0, 0));for(uint8_t i = 0; i < 3; ++i)// 將直方圖歸一化到范圍 [ 0, histImage.rows ]normalize(RGB_Hist[i], RGB_Hist[i], 0, histImage.rows, NORM_MINMAX, -1, Mat());// 在直方圖畫(huà)布上畫(huà)出直方圖for (int i = 1; i < histSize; i++){line(histImage, Point(bin_w*(i - 1), hist_h - cvRound(RGB_Hist[0].at<float>(i - 1))),Point(bin_w*(i), hist_h - cvRound(RGB_Hist[0].at<float>(i))),Scalar(0, 0, 255), 2, 8, 0);line(histImage, Point(bin_w*(i - 1), hist_h - cvRound(RGB_Hist[1].at<float>(i - 1))),Point(bin_w*(i), hist_h - cvRound(RGB_Hist[1].at<float>(i))),Scalar(0, 255, 0), 2, 8, 0);line(histImage, Point(bin_w*(i - 1), hist_h - cvRound(RGB_Hist[2].at<float>(i - 1))),Point(bin_w*(i), hist_h - cvRound(RGB_Hist[2].at<float>(i))),Scalar(255, 0, 0), 2, 8, 0);}/// 顯示直方圖///imshow("calcHist Demo", histImage);return histImage; }Mat getEqualizeHistImage(Mat input) {int gray[256] = { 0 }; //記錄每個(gè)灰度級(jí)別下的像素個(gè)數(shù)double gray_prob[256] = { 0 }; //記錄灰度分布密度double gray_distribution[256] = { 0 }; //記錄累計(jì)密度int gray_equal[256] = { 0 }; //均衡化后的灰度值int gray_sum = 0; //像素總數(shù)Mat chRGB[3];split(input, chRGB);Mat output = input.clone();for (unsigned char k = 0; k < 3; ++k){for (unsigned short i = 0; i < 256; ++i){gray[i] = 0;gray_prob[i] = 0;gray_distribution[i] = 0;gray_equal[i] = 0;}gray_sum = chRGB[k].cols * chRGB[k].rows;//統(tǒng)計(jì)每個(gè)灰度下的像素個(gè)數(shù)for (int i = 0; i < chRGB[k].rows; i++){uchar* p = chRGB[k].ptr<uchar>(i);for (int j = 0; j < chRGB[k].cols; j++){int vaule = p[j];gray[vaule]++;}}//統(tǒng)計(jì)灰度頻率for (int i = 0; i < 256; i++){gray_prob[i] = ((double)gray[i] / gray_sum);}//計(jì)算累計(jì)密度gray_distribution[0] = gray_prob[0];for (int i = 1; i < 256; i++){gray_distribution[i] = gray_distribution[i - 1] + gray_prob[i];}//重新計(jì)算均衡化后的灰度值，四舍五入。參考公式：(N-1)*T+0.5for (int i = 0; i < 256; i++){gray_equal[i] = (uchar)(255 * gray_distribution[i] + 0.5);}//直方圖均衡化,更新原圖每個(gè)點(diǎn)的像素值for (int i = 0; i < chRGB[k].rows; i++){uchar* p = chRGB[k].ptr<uchar>(i);for (int j = 0; j < chRGB[k].cols; j++){p[j] = gray_equal[p[j]];}}}merge(chRGB, 3, output);return output; }int main(int argc, char * argv[]) {Mat Image = imread("D:\\Work\\OpenCV\\Workplace\\Test_1\\1.jpg");imshow("原圖", Image);imshow("原圖直方圖", getHistImage(Image));imshow("直方圖均衡化后的圖像", getEqualizeHistImage(Image));imshow("直方圖均衡化后的直方圖", getHistImage(getEqualizeHistImage(Image)));waitKey(0);return 0; }

總結(jié)

以上是生活随笔為你收集整理的【机器视觉学习笔记】直方图的绘制及直方图均衡化（C++）的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。