对象检测与跟踪(基于颜色)

青旅半醒 2022-05-24 09:39 193阅读 0赞

![70][]

第一步：inRange过滤：

![70 1][]

第二步：形态学操作提取：

![70 2][]

第三步：做个膨胀处理：

![70 3][]

对原视频流中做外接矩形获取，以及位置标定

![70 4][]

代码处理如下：

#include <opencv2/opencv.hpp>
    #include <iostream>
    
    using namespace std;
    using namespace cv;
    
    //外接矩形获取
    Rect roi;
    void processFrame(Mat &binary, Rect &rect);
    
    int main(int argc, char* argv) {
    	// load video
    	VideoCapture capture;
    	capture.open("F:/video_006.mp4");
    	if (!capture.isOpened()) {
    		printf("could not find video file");
    		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", CV_WINDOW_AUTOSIZE);
    	namedWindow("track mask", CV_WINDOW_AUTOSIZE);
    	while (capture.read(frame)) {
    
    
    		//1 inRange过滤
    		inRange(frame, Scalar(0, 127, 0), Scalar(120, 255, 120), mask); // 过滤
    		//2 形态学操作提取
    		morphologyEx(mask, mask, MORPH_OPEN, kernel1, Point(-1, -1), 1); // 开操作
    		//3 轮廓查找
    		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);
    
    		// trigger exit
    		char c = waitKey(100);
    		if (c == 27) {
    			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) {
    		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;
    	}
    
    }

[70]: /images/20220524/2213fe020cb64739811d16e3b7744086.png
[70 1]: /images/20220524/137dcdfbb6f84b19a588fd39ae236df9.png
[70 2]: /images/20220524/1ecbbb379a794a96894ebf54c41575a4.png
[70 3]: /images/20220524/bb03fc343d3a4b54b8a2871b697dbda8.png
[70 4]: /images/20220524/a37f47e9fcc243258f66b47ffcf1aa6d.png