Opencv --- 基于颜色的对象检测与跟踪

1. 基于颜色跟踪实现步骤：

        inRange过滤

        形态学操作提取

        轮廓查找

        外接矩形获取

        位置标定

2. 代码演示：

#include<iostream>
#include<opencv2/opencv.hpp>

using namespace std;
using namespace cv;

Rect roi;
void processFrame(Mat &binary,Rect &rect);

int main(int argc, char** argv)
{
	VideoCapture capture;
	capture.open("E:/技能学习/Opencv视频分析与对象跟踪/video_006.mp4");
	if (!capture.isOpened())
	{
		cout << "could not find the video file..." << endl;
		return -1;
	}

	Mat frame, mask;
	Mat kernel1 = getStructuringElement(MORPH_RECT, Size(3, 3), Point(-1, -1));
	Mat kernel2 = getStructuringElement(MORPH_RECT, Size(5, 5), Point(-1, -1));

	namedWindow("input video", WINDOW_AUTOSIZE);
	namedWindow("track mask", WINDOW_AUTOSIZE);

	while (capture.read(frame))
	{
		//过滤
		inRange(frame, Scalar(0, 127, 0), Scalar(120, 255, 120), mask);

		//开操作
		morphologyEx(mask, mask, MORPH_OPEN, kernel1, Point(-1, -1), 1);

		//膨胀
		dilate(mask, mask, kernel2, Point(-1, -1), 4);

		//获取最大外接矩形与位置标定
		processFrame(mask, roi);

		rectangle(frame, roi, Scalar(0, 0, 255), 3, 8, 0);

		imshow("input video", frame);
		imshow("track mask", mask);
		
		char c = waitKey(100);
		if (c == 27)
		{
			break;
		}
	}

	capture.release();

	waitKey(0);
	destroyAllWindows();
	return 0;
}

void processFrame(Mat &binary, Rect &rect)
{
	vector<vector<Point>>contours;
	vector<Vec4i>hireachy;
	 
	//查找轮廓  CV_RETR_EXTERNAL：只有最外层轮廓；CHAIN_APPROX_SIMPLE：压缩轮廓，把横竖撇捺都压缩得只剩下顶点；
	findContours(binary, contours, hireachy, RETR_EXTERNAL, CHAIN_APPROX_SIMPLE, Point(0, 0));
	if (contours.size() > 0)
	{
		double maxArea = 0.0;
		for (size_t t = 0; t < contours.size(); t++)
		{
			//contourArea()函数的作用：主要用于计算图像轮廓的面积
			double area = contourArea(contours[static_cast<int>(t)]);

			if (area > maxArea)
			{
				maxArea = area;
				//boundingRect函数作用：计算轮廓的垂直边界最小矩形，矩形是与图像上下边界平行的
				rect = boundingRect(contours[static_cast<int>(t)]);
			}
		}
	}
	else
	{
		rect.x = rect.y = rect.width = rect.height = 0;
	}

}