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;
}
}