颜色识别opencv+trackbar调节目标颜色hsv的阈值
颜色识别,trackbar调节颜色阈值
#include <iostream>
#include "opencv2/highgui/highgui.hpp"
#include "opencv2/imgproc/imgproc.hpp"
using namespace cv;
using namespace std;
Mat imgOriginal;
int iLowH = 100;
int iHighH = 140;
int iLowS = 90;
int iHighS = 255;
int iLowV = 90;
int iHighV = 255;
void callback(int, void*);
int main(int argc, char** argv)
{
namedWindow("Control", CV_WINDOW_AUTOSIZE); //create a window called "Control"
imgOriginal = imread("solder1.jpg");
resize(imgOriginal, imgOriginal, Size(400, 300));
imshow("Original", imgOriginal); //show the original image
//Create trackbars in "Control" window
createTrackbar("LowH", "Control", &iLowH, 179, on_mouse); //Hue (0 - 179)
createTrackbar("HighH", "Control", &iHighH, 179, on_mouse);
createTrackbar("LowS", "Control", &iLowS, 255, on_mouse); //Saturation (0 - 255)
createTrackbar("HighS", "Control", &iHighS, 255, on_mouse);
createTrackbar("LowV", "Control", &iLowV, 255, on_mouse); //Value (0 - 255)
createTrackbar("HighV", "Control", &iHighV, 255, on_mouse);
waitKey(0);
return 0;
}
void callback(int ,void*)
{
Mat imgHSV;
vector<Mat> hsvSplit;
cvtColor(imgOriginal, imgHSV, COLOR_BGR2HSV); //Convert the captured frame from BGR to HSV
//因为我们读取的是彩色图,直方图均衡化需要在HSV空间做
split(imgHSV, hsvSplit);
equalizeHist(hsvSplit[2], hsvSplit[2]);
merge(hsvSplit, imgHSV);
Mat imgThresholded;
inRange(imgHSV, Scalar(iLowH, iLowS, iLowV), Scalar(iHighH, iHighS, iHighV), imgThresholded); //Threshold the image
//开操作 (去除一些噪点)
Mat element = getStructuringElement(MORPH_RECT, Size(5, 5));
morphologyEx(imgThresholded, imgThresholded, MORPH_OPEN, element);
//闭操作 (连接一些连通域)
morphologyEx(imgThresholded, imgThresholded, MORPH_CLOSE, element);
imshow("Thresholded Image", imgThresholded); //show the thresholded image
}