Opencv2.4学习::颜色提取

颜色提取


 Opencv中,颜色提取的一种方式是将BGR空间下的图像转换为HSV空间下,然后利用opencv自带函数inRange,设置需提取的HSV各分量上下限,从而进行提取。


关于颜色空间

关于颜色空间转换请参见:https://blog.csdn.net/dieju8330/article/details/82465616

以及

https://blog.csdn.net/ColdWindHA/article/details/82080176#%E4%BA%8C%E3%80%81HSV%E5%80%BC%E5%AF%B9%E9%A2%9C%E8%89%B2%E7%9A%84%E5%BD%B1%E5%93%8D

下给出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,到相应数值,即可提取到相应部分。

猜你喜欢

转载自blog.csdn.net/dieju8330/article/details/82775451
今日推荐