1. 程式人生 > >Opencv2.4學習::顏色提取

Opencv2.4學習::顏色提取

顏色提取

 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,到相應數值,即可提取到相應部分。