颜色提取
Opencv中,颜色提取的一种方式是将BGR空间下的图像转换为HSV空间下,然后利用opencv自带函数inRange,设置需提取的HSV各分量上下限,从而进行提取。
关于颜色空间
关于颜色空间转换请参见:https://blog.csdn.net/dieju8330/article/details/82465616
以及
下给出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,到相应数值,即可提取到相应部分。