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

来源:互联网 发布:php金字塔代码 编辑:程序博客网 时间:2024/06/05 08:22

最近在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;}


效果如下图:



原创粉丝点击