[OpenCV新手向]一个OpenCV颜色空间转化的实例

最近在OpenCV中文论坛上解答了个问题，大概问题是这样的，把下图中绿色的八卦部分抠出来：

可以看出问题解决方案很直接：遍历图片的每个像素，然后如果像素的颜色接近于绿色，保留此像素；反之遗弃，设值为0。解决思路跟photoshop的颜色选择功能类似。

问题主要的难点在于如何去比较颜色。输入图像的RGB色彩并不适用于颜色的比较；这里就要引入HSV色彩空间，把RGB颜色转化成H色相，S饱和度，V色调（亮度）。根据饼状图，我们发现绿色范围（转化成0~255区间）大概在35到90之间。所以，遍历过程中，我们留下H通道值在35~90范围内，并且饱和度和色调足够高（保证像素足够明亮）的像素；反之，剩下的我们认为是背景和噪点。

代码如下：

#include <opencv2/opencv.hpp>
#include <opencv2/highgui/highgui.hpp>

using namespace cv;
void getGreenMask(Mat& src, Mat& dst)
{
	cvtColor(src, dst, cv::COLOR_BGR2HSV);
	int hi = 90, lo = 35;
	int lo_sat_v = 50;
	for(int i = 0; i < src.rows; i ++)
	{
		Vec3b * row_ptr = dst.ptr<Vec3b>(i);
		for(int j = 0; j < src.cols; j ++)
		{
			Vec3b hsv = row_ptr[j];
			if((hsv[1] < lo_sat_v || hsv[2] < lo_sat_v) // 饱和度和亮度过低
				|| (hsv[0] > hi || hsv[0] < lo))        // 色相不在范围内
			{
				row_ptr[j] = Vec3b::all(0);             // 剔除不在范围内的像素
			}
		}
	}
	cvtColor(dst, dst, cv::COLOR_HSV2BGR);
}
int main(int argc, char* argv[])
{
	Mat in = imread("F:/test.JPG");
	namedWindow("green channel");
	Mat out(in.size(), CV_8UC3, Scalar::all(0));
	getGreenMask(in, out);
	imshow("green channel", out);
	waitKey();
    return 0;
}
           

效果如下图：