C++ OpenCV單峰三角閾值法Thresh_Unimodal詳解
需求說明
在對(duì)圖像進(jìn)行處理時(shí)，經(jīng)常會(huì)有這類需求：想通過閾值對(duì)圖像進(jìn)行二值化分割，以提取自己感興趣的區(qū)域，常見的閾值分割方法有常數(shù)分割、最大類間方差法、雙峰分割、三角法等等，不同的場(chǎng)景應(yīng)用不同的閾值方法。
今天要講的方法，適合當(dāng)圖像的直方圖具有明顯單峰特征時(shí)使用，結(jié)合了三角法的原理而設(shè)計(jì)，相比較OpenCV自帶的三角法，好處是可以根據(jù)自身需求合理修改函數(shù)；如果用OpenCV庫的函數(shù)，只有一個(gè)接口，若不能達(dá)到較理想的應(yīng)用效果，就束手無策了。
下面介紹具體實(shí)現(xiàn)流程。
具體流程
1）取圖像的灰度圖，并遍歷統(tǒng)計(jì)0-255各個(gè)灰度值所出現(xiàn)的次數(shù)。
cv::Mat src = imread("test.jpg", 0);
cv::Mat hist = cv::Mat::zeros(1, 256, CV_32FC1);
for (int i = 0; i < src.rows; ++i)
{
for (int j = 0; j < src.cols; ++j)
{
hist.at<float>(0, src.at <uchar>(i, j))++;
}
}
2）去除0和255的直方圖數(shù)據(jù)，這一步就是OpenCV三角法所沒有的。很多人可能不理解為什么要這一步，在你對(duì)圖像進(jìn)行閾值化時(shí)如果提前進(jìn)行了相關(guān)的運(yùn)算，可能導(dǎo)致結(jié)果大于255的數(shù)值全部變?yōu)?55，或者數(shù)值低于0的數(shù)值全部變?yōu)?，這就使得0和255的數(shù)值其實(shí)涵蓋了許多數(shù)值，呈累加態(tài)，很容易形成雙峰，這樣就很難找到我們真正想要的峰。例如0和255的數(shù)值都是10000左右,0略大一些，而我們的真峰是在250左右的灰度值，數(shù)值只有8000多，那么在后續(xù)閾值計(jì)算時(shí)就會(huì)因?yàn)榉宓姆较蝈e(cuò)了而帶來毀滅性打擊。別覺得我說夸張了，只有自己去碰碰壁才能深刻領(lǐng)悟我說的。
hist.at<float>(0, 255) = 0; hist.at<float>(0, 0) = 0;
3）確認(rèn)峰值位置，maxidx是峰值對(duì)應(yīng)的灰度值，max是峰值高度，也是灰度值對(duì)應(yīng)數(shù)據(jù)的個(gè)數(shù)。
float max = 0;
int maxidx = 0;
for (int i = 0; i < 256; ++i)
{
if (hist.at<float>(0, i) > max)
{
max = hist.at<float>(0, i);
maxidx = i;
}
}
4）判斷峰值在左側(cè)還是右側(cè)，true為左側(cè)，false為右側(cè)。
bool lr = maxidx < 127;
5）當(dāng)在左側(cè)時(shí)，連接峰值(maxidx,max)和(255,0)點(diǎn)，用兩點(diǎn)建立直線公式，如下圖所示公式。 L的表達(dá)式可以轉(zhuǎn)換為Ax+By+C=0的形式，A是-max，B是maxidx-255，C是max*255，在結(jié)合距離公式可以計(jì)算出直方圖曲線上每個(gè)點(diǎn)到直線的距離，取距離最長的那個(gè)點(diǎn)作為閾值。
if (lr)
{
float A = float(-max);
float B = float(maxidx - 255);
float C = float(max * 255);
for (int i = maxidx + 1; i < 256; ++i)
{
float x0 = float(i);
float y0 = hist.at<float>(0, i);
float d = abs(A * x0 + B * y0 + C) / std::sqrt(A * A + B * B);
if (d > maxd)
{
maxd = d;
maxdidx = i;
}
}
}
6）右側(cè)同理，連接峰值(maxidx,max)和(0,0)點(diǎn)，公式ABC如代碼所示。
else {
float A = float(-max);
float B = float(maxidx);
float C = 0.0f;
for (int i = 0; i < maxidx; ++i)
{
float x0 = float(i);
float y0 = hist.at<float>(0, i);
float d = abs(A * x0 + B * y0 + C) / std::sqrt(A * A + B * B);
if (d > maxd)
{
maxd = d;
maxdidx = i;
}
}
}
7）二值化，完成。
result.setTo(255, src > maxdidx); idx = maxdidx; return result;
功能函數(shù)
// 單峰三角閾值法
cv::Mat Thresh_Unimodal(cv::Mat &src, int& idx)
{
cv::Mat result = cv::Mat::zeros(src.size(), CV_8UC1);
// 統(tǒng)計(jì)直方圖
cv::Mat hist = cv::Mat::zeros(1, 256, CV_32FC1);
for (int i = 0; i < src.rows; ++i)
{
for (int j = 0; j < src.cols; ++j)
{
hist.at<float>(0, src.at<uchar>(i, j))++;
}
}
hist.at<float>(0, 255) = 0;
hist.at<float>(0, 0) = 0;
// 搜索最大值位置
float max = 0;
int maxidx = 0;
for (int i = 0; i < 256; ++i)
{
if (hist.at<float>(0, i) > max)
{
max = hist.at<float>(0, i);
maxidx = i;
}
}
// 判斷最大點(diǎn)在哪一側(cè)，true為左側(cè)，false為右側(cè)
bool lr = maxidx < 127;
float maxd = 0;
int maxdidx = 0;
// 假設(shè)在左側(cè)
if (lr)
{
float A = float(-max);
float B = float(maxidx - 255);
float C = float(max * 255);
for (int i = maxidx + 1; i < 256; ++i)
{
float x0 = float(i);
float y0 = hist.at<float>(0, i);
float d = abs(A * x0 + B * y0 + C) / std::sqrt(A * A + B * B);
if (d > maxd)
{
maxd = d;
maxdidx = i;
}
}
}
// 假設(shè)在右側(cè)
else {
float A = float(-max);
float B = float(maxidx);
float C = 0.0f;
for (int i = 0; i < maxidx; ++i)
{
float x0 = float(i);
float y0 = hist.at<float>(0, i);
float d = abs(A * x0 + B * y0 + C) / std::sqrt(A * A + B * B);
if (d > maxd)
{
maxd = d;
maxdidx = i;
}
}
}
// 二值化
result.setTo(255, src > maxdidx);
idx = maxdidx;
return result;
}
C++測(cè)試代碼
#include <iostream>
#include <time.h>
#include <opencv2/opencv.hpp>
using namespace std;
using namespace cv;
cv::Mat DrawHistImg(cv::Mat &hist);
cv::Mat Thresh_Unimodal(cv::Mat &src, int& idx);
int main()
{
cv::Mat src = imread("test.jpg", 0);
// 繪制均衡化后直方圖
cv::Mat hrI = DrawHistImg(src);
// 單峰三角閾值法
int thresh;
cv::Mat result = Thresh_Unimodal(src, thresh);
cout << " thresh: " << thresh << endl;
imshow("original", src);
imshow("hist", hrI);
imshow("result", result);
waitKey(0);
return 0;
}
// 繪制簡易直方圖
cv::Mat DrawHistImg(cv::Mat &src)
{
cv::Mat hist = cv::Mat::zeros(1, 256, CV_32FC1);
for (int i = 0; i < src.rows; ++i)
{
for (int j = 0; j < src.cols; ++j)
{
hist.at<float>(0, src.at <uchar>(i, j))++;
}
}
cv::Mat histImage = cv::Mat::zeros(540, 1020, CV_8UC1);
const int bins = 255;
double maxValue;
cv::Point2i maxLoc;
cv::minMaxLoc(hist, 0, &maxValue, 0, &maxLoc);
int scale = 4;
int histHeight = 540;
for (int i = 0; i < bins; i++)
{
float binValue = (hist.at<float>(i));
int height = cvRound(binValue * histHeight / maxValue);
cv::rectangle(histImage, cv::Point(i * scale, histHeight),
cv::Point((i + 1) * scale - 1, histHeight - height), cv::Scalar(255), -1);
}
return histImage;
}
// 單峰三角閾值法
cv::Mat Thresh_Unimodal(cv::Mat &src, int& idx)
{
cv::Mat result = cv::Mat::zeros(src.size(), CV_8UC1);
// 統(tǒng)計(jì)直方圖
cv::Mat hist = cv::Mat::zeros(1, 256, CV_32FC1);
for (int i = 0; i < src.rows; ++i)
{
for (int j = 0; j < src.cols; ++j)
{
hist.at<float>(0, src.at<uchar>(i, j))++;
}
}
hist.at<float>(0, 255) = 0;
hist.at<float>(0, 0) = 0;
// 搜索最大值位置
float max = 0;
int maxidx = 0;
for (int i = 0; i < 256; ++i)
{
if (hist.at<float>(0, i) > max)
{
max = hist.at<float>(0, i);
maxidx = i;
}
}
// 判斷最大點(diǎn)在哪一側(cè)，true為左側(cè)，false為右側(cè)
bool lr = maxidx < 127;
float maxd = 0;
int maxdidx = 0;
// 假設(shè)在左側(cè)
if (lr)
{
float A = float(-max);
float B = float(maxidx - 255);
float C = float(max * 255);
for (int i = maxidx + 1; i < 256; ++i)
{
float x0 = float(i);
float y0 = hist.at<float>(0, i);
float d = abs(A * x0 + B * y0 + C) / std::sqrt(A * A + B * B);
if (d > maxd)
{
maxd = d;
maxdidx = i;
}
}
}
// 假設(shè)在右側(cè)
else {
float A = float(-max);
float B = float(maxidx);
float C = 0.0f;
for (int i = 0; i < maxidx; ++i)
{
float x0 = float(i);
float y0 = hist.at<float>(0, i);
float d = abs(A * x0 + B * y0 + C) / std::sqrt(A * A + B * B);
if (d > maxd)
{
maxd = d;
maxdidx = i;
}
}
}
// 二值化
result.setTo(255, src > maxdidx);
idx = maxdidx;
return result;
}
測(cè)試效果
圖1 原圖灰度圖
圖2 直方圖
圖3 閾值圖
圖4 閾值結(jié)果
通過imagewatch插件可以觀察閾值203是不是在距離最遠(yuǎn)的位置，答案是肯定的。
如果函數(shù)有什么可以改進(jìn)完善的地方，非常歡迎大家指出，一同進(jìn)步何樂而不為呢~?
以上就是C++ OpenCV單峰三角閾值法Thresh_Unimodal詳解的詳細(xì)內(nèi)容，更多關(guān)于C++ OpenCV單峰三角閾值法的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
關(guān)于C++靜態(tài)成員函數(shù)訪問非靜態(tài)成員變量的問題
靜態(tài)成員函數(shù)不能訪問非靜態(tài)成員，這是因?yàn)殪o態(tài)函數(shù)屬于類而不是屬于整個(gè)對(duì)象，靜態(tài)函數(shù)中的 member可能都沒有分配內(nèi)存。靜態(tài)成員函數(shù)沒有隱含的this自變量。所以，它就無法訪問自己類的非靜態(tài)成員2013-10-10
這篇文章主要為大家詳細(xì)介紹了C語言制作貪吃蛇小游戲，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2022-07-07
運(yùn)算符重載函數(shù)可以是類的成員函數(shù)，也可以是類的友元函數(shù)，還可以是既非類的成員函數(shù)也不是友元函數(shù)的普通函數(shù)2013-10-10
這篇文章主要介紹了C++設(shè)計(jì)模式之建造者模式,一個(gè)復(fù)雜對(duì)象是由多個(gè)部件組成的,建造者模式是把復(fù)雜對(duì)象的創(chuàng)建和部件的創(chuàng)建分別開來,分別用Builder類和Director類來表示,需要的朋友可以參考下2014-09-09
Qt實(shí)現(xiàn)指針式時(shí)鐘 Qt實(shí)現(xiàn)動(dòng)態(tài)時(shí)鐘
這篇文章主要為大家詳細(xì)介紹了Qt實(shí)現(xiàn)指針式時(shí)鐘，Qt實(shí)現(xiàn)動(dòng)態(tài)時(shí)鐘，兩者相互切換，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2020-07-07
一個(gè)快速排序算法代碼一個(gè)快速排序算法代碼，代碼內(nèi)有注釋,大家參考使用吧2014-01-01
C語言編程數(shù)據(jù)結(jié)構(gòu)基礎(chǔ)詳解小白篇
這篇文章主要介紹了數(shù)據(jù)結(jié)構(gòu)的基礎(chǔ)，非常適合初學(xué)數(shù)據(jù)結(jié)構(gòu)的小白，有需要的朋友可以借鑒參考下，希望可以有所幫助，祝大家多多進(jìn)步，早日升職加薪2021-09-09
Linux?C/C++實(shí)現(xiàn)顯示NIC流量統(tǒng)計(jì)信息
NIC流量統(tǒng)計(jì)信息是由操作系統(tǒng)維護(hù)的,當(dāng)數(shù)據(jù)包通過NIC傳輸時(shí),操作系統(tǒng)會(huì)更新相關(guān)的計(jì)數(shù)器,通過讀取這些計(jì)數(shù)器,我們可以獲得關(guān)于網(wǎng)絡(luò)流量的信息,下面我們就來學(xué)習(xí)一下如何通過C/C++實(shí)現(xiàn)顯示NIC流量統(tǒng)計(jì)信息吧2024-01-01
文章介紹了C++中向Lambda表達(dá)式傳遞參數(shù)和使用捕獲列表的基本規(guī)則和示例,參數(shù)傳遞要求嚴(yán)格匹配類型和數(shù)量,而捕獲列表允許Lambda訪問外部變量,并可以按值或引用捕獲,文章還討論了捕獲的陷阱和解決方案,以及參數(shù)傳遞與捕獲列表的區(qū)別和應(yīng)用場(chǎng)景,感興趣的朋友一起看看吧2025-03-03
這篇文章主要為大家詳細(xì)介紹了在C/C++中如何寫調(diào)試宏的相關(guān)知識(shí),文中的示例代碼講解詳細(xì),具有一定的借鑒價(jià)值,有需要的小伙伴可以參考一下2024-05-05