作者:RayChiu_Labloy
版权声明:著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处
目录
背景需求
我们最终要提取图片中的指定的颜色,大多数都是利用HSV色彩空间的上下限范围值, 然后利用inRange函数实现功能。
官方文档给出的inRange函数解释:
代码
#include<opencv2/core.hpp>
#include<opencv2/highgui.hpp>
#include<opencv2/imgproc.hpp>
using namespace cv;
#include<iostream>
#include<string>
using namespace std;
//输入图像
Mat img;
//灰度值归一化
Mat bgr;
//HSV图像
Mat hsv;
//色相
int hmin = 0;
int hmin_Max = 360;
int hmax = 360;
int hmax_Max = 360;
//饱和度
int smin = 0;
int smin_Max = 255;
int smax = 255;
int smax_Max = 255;
//亮度
int vmin = 106;
int vmin_Max = 255;
int vmax = 250;
int vmax_Max = 255;
//显示原图的窗口
string windowName = "src";
//输出图像的显示窗口
string dstName = "dst";
//输出图像
Mat dst;
//回调函数
void callBack(int, void*)
{
//输出图像分配内存
dst = Mat::zeros(img.size(), CV_32FC3);
//掩码
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);
//只保留
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<Vec3f>(r, c) = bgr.at<Vec3f>(r, c);
}
}
}
//使用开操作和闭操作去除二值化图像中的噪声
Mat element = getStructuringElement(MORPH_RECT, Size(5, 5));
morphologyEx(dst, dst, MORPH_OPEN, element);
morphologyEx(dst, dst, MORPH_CLOSE, element);
//输出图像
imshow(dstName, dst);
//保存图像
dst.convertTo(dst, CV_8UC3, 255.0, 0);
imwrite("HSV_inRange.jpg", dst);
}
int main()
{
//输入图像
img = imread("E:/Users/raychiu/Desktop/Industrial defect detection data/Handle classification/1.jpg", IMREAD_COLOR);
if (!img.data || img.channels() != 3)
return -1;
resize(img, img, Size(432, 432), 0, 0, INTER_LINEAR_EXACT);
imshow(windowName, img);
//彩色图像的灰度值归一化
img.convertTo(bgr, CV_32FC3, 1.0 / 255, 0);
//颜色空间转换
cvtColor(bgr, hsv, COLOR_BGR2HSV);
//vector<Mat> hsvSplit;
//split(hsv, hsvSplit);
//equalizeHist(hsvSplit[1], hsvSplit[1]); //对HSV的亮度通道进行直方图均衡
//merge(hsvSplit, hsv); //合并三种通道
//cvtColor(hsv, img, COLOR_HSV2BGR); //将HSV空间转回至RGB空间,为接下来的颜色识别做准备
//定义输出图像的显示窗口
namedWindow(dstName, WINDOW_GUI_EXPANDED);
//调节色相 H
createTrackbar("hmin", dstName, &hmin, hmin_Max, callBack);
createTrackbar("hmax", dstName, &hmax, hmax_Max, callBack);
//调节饱和度 S
createTrackbar("smin", dstName, &smin, smin_Max, callBack);
createTrackbar("smax", dstName, &smax, smax_Max, callBack);
//调节亮度 V
createTrackbar("vmin", dstName, &vmin, vmin_Max, callBack);
createTrackbar("vmax", dstName, &vmax, vmax_Max, callBack);
callBack(0, 0);
waitKey(0);
return 0;
}
实验图片和效果
参考:OpenCV-利用函数inRange进行颜色分割(针对HSV颜色空间)_我们的时光!-CSDN博客_inrange函数取出红色
【如果对您有帮助,交个朋友给个一键三连吧,您的肯定是我博客高质量维护的动力!!!】