Opencv2.4學習::顏色提取
阿新 • • 發佈:2018-12-10
顏色提取
Opencv中,顏色提取的一種方式是將BGR空間下的影象轉換為HSV空間下,然後利用opencv自帶函式inRange,設定需提取的HSV各分量上下限,從而進行提取。
關於顏色空間
以及
下給出HSV對應BGR顏色的表格 (下面會用到):
黑 |
灰 |
白 |
紅 |
橙 |
黃 |
綠 |
青 |
藍 |
紫 |
||
hmin |
0 |
0 |
0 |
0 |
156 |
11 |
26 |
35 |
78 |
100 |
125 |
hmax |
180 |
180 |
180 |
10 |
180 |
25 |
34; |
77 |
99 |
124 |
155 |
smin |
0 |
0 |
0 |
43 |
43 |
43 |
43 |
43 |
43 |
43 |
|
smax |
255 |
43 |
30 |
255 |
255 |
255 |
255 |
255 |
255 |
255 |
|
vmin |
0 |
46 |
221 |
46 |
46 |
46 |
46 |
46 |
46 |
46 |
|
vmax |
46 |
220 |
255 |
255 |
255 |
255 |
255 |
255 |
255 |
255 |
當然,這只是個大致範圍。
核心函式:
void inRange(InputArray src, InputArray lowerb, InputArray upperb, OutputArray mask);
- src:輸入影象
- lowerb:若輸入的為三通道hsv影象,則為提取的下限,應填入Scalar(hmin,smin,vmin)
- upperb:若輸入的為三通道hsv影象,則為提取顏色的上限,應填入Scalar(hmax,smax,vmax)
- mask:輸出為一個二值化的影象,提取出來的部分為255,其餘部分為0
程式碼實現:
#include<opencv2/core/core.hpp>
#include<opencv2/highgui/highgui.hpp>
#include<opencv2/imgproc/imgproc.hpp>
#include<iostream>
#include<string>
using namespace std;
using namespace cv;
//bgr影象
Mat bgr;
//HSV影象
Mat hsv;
//色相
int hmin = 0;//h分量提取下限
int hmax = 180;//h分量提取上限
int h_Max = 180; //h分量可取的最大值
//飽和度
int smin = 0;//s分量提取下限
int smax = 255;//s分量提取上限
int s_Max = 255;//s分量可取的最大值
//亮度
int vmin = 0;//v分量提取下限
int vmax = 255;//v分量提取上限
int v_Max = 255;//v分量可取的最大值
//顯示原圖的視窗
string windowName = "src";
//輸出影象的顯示視窗
string dstName = "dst";
//輸出影象
Mat dst;
//回撥函式
void callBack(int, void*)
{
//輸出影象分配記憶體
dst = Mat::zeros(bgr.size(), bgr.type());
//遮罩
Mat mask;
//inRange(hsv, Scalar(hmin, smin / float(smin_Max), vmin / float(vmin_Max)), Scalar(hmax, smax / float(smax_Max), vmax / float(vmax_Max)), mask);
inRange(hsv, Scalar(hmin, smin, vmin), Scalar(hmax, smax, vmax), mask);
//只保留
for (int r = 0; r < bgr.rows; r++)
{
for (int c = 0; c < bgr.cols; c++)
{
if (mask.at<uchar>(r, c) == 255)
{
dst.at<Vec3b>(r, c)[0] = bgr.at<Vec3b>(r, c)[0];
dst.at<Vec3b>(r, c)[1] = bgr.at<Vec3b>(r, c)[1];
dst.at<Vec3b>(r, c)[2] = bgr.at<Vec3b>(r, c)[2];
}
}
}
imshow("mask", mask);
//輸出影象
imshow("dst", dst);
}
int main()
{
//輸入影象
Mat srcImage = imread("F:\\opencv_re_learn\\chepai.jpg");
if (!srcImage.data){
cout << "falied to read" << endl;
system("pause");
return -1;
}
imshow(windowName, srcImage);
bgr = srcImage;
//顏色空間轉換
cvtColor(bgr, hsv, COLOR_BGR2HSV);
//定義輸出影象的顯示視窗
namedWindow(dstName, CV_WINDOW_AUTOSIZE);
//調節色相 H
createTrackbar("hmin", dstName, &hmin, h_Max, callBack);
createTrackbar("hmax", dstName, &hmax, h_Max, callBack);
//調節飽和度 S
createTrackbar("smin", dstName, &smin, s_Max, callBack);
createTrackbar("smax", dstName, &smax, s_Max, callBack);
//調節亮度 V
createTrackbar("vmin", dstName, &vmin, v_Max, callBack);
createTrackbar("vmax", dstName, &vmax, v_Max, callBack);
callBack(0, 0);
waitKey(0);
return 0;
}
實現效果:
示例中,目的是提取車牌的藍色部分,因此先查HSV表,有:
所以,直接拖動tracebar,到相應數值,即可提取到相應部分。