一个OpenCV颜色空间转化的实例

2012 年 07 月 10 日
learning
OpenCV

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

Input

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

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

代码如下:

#include <opencv2/opencv.hpp>  
#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;  
}

效果如下图： Input