创 OpenCV - 视频分析与对象跟踪实战（03. 对象检测与跟踪（基于颜色））

对象检测与跟踪（基于颜色）

• 利用颜色范围过滤（基于颜色跟踪实现步骤）
  • inRange过滤
  • 形态学操作提取
  • 轮廓查找
  • 外接矩形获取
  • 位置标定
• 标注与测量

代码演示

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

using namespace cv;
using namespace std;

// 绘制感兴趣区域
Rect roi;
void processFrame(Mat &binary, Rect &rect);// & 表示引用，形参改动时，实参也会改动

int main(int argc, char** argv){
	VideoCapture capture;
	capture.open("E:/Experiment/OpenCV/src/video_006.mp4"); 

	if (!capture.isOpened()){
		printf("Video could not load ...."); 
		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("test-show", CV_WINDOW_AUTOSIZE);
	namedWindow("track mask", CV_WINDOW_AUTOSIZE);
	namedWindow("output video", CV_WINDOW_AUTOSIZE);
	while (capture.read(frame)){
		imshow("test-show", 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);
		imshow("track mask", mask);

		processFrame(mask, roi); //轮廓发现与位置标定
		rectangle(frame, roi, Scalar(0, 0, 255), 3, 8, 0);
		imshow("output video", frame);

		char c = waitKey(100); // 每隔100毫秒读取一帧
		if (c == 27){ //按Esc键可退出播放
			break;
		}
	}
	capture.release(); // 释放资源
	waitKey(0);
	return 0;
}

// 轮廓发现与位置标定
void processFrame(Mat &binary, Rect &rect){
	vector<vector<Point>> contours;
	vector<Vec4i> hireachy;
	findContours(binary, contours, hireachy, RETR_EXTERNAL, CHAIN_APPROX_SIMPLE, Point(0, 0));

	if (contours.size() > 0){// 如果找到轮廓数>0个
		double maxArea = 0.0;
		for (size_t t = 0; t < contours.size(); t++){
			double area = contourArea(contours[static_cast<int>(t)]);
			if (area > maxArea){
				maxArea = area;
				rect = boundingRect(contours[static_cast<int>(t)]);
			}
		}
	}else{
		rect.x = rect.y = rect.width = rect.height = 0;
	}
}