opencv：物体跟踪

蔚落 2022-12-10 04:57 140阅读 0赞

> 当前使用版本：opencv4.3

#include <string>
    #include <iostream>
    #include <opencv2/highgui/highgui.hpp>
    #include <opencv2/tracking.hpp>
    
    using namespace std;
    using namespace cv;
    
    
    int main()
    {
        
    
        string filename = "/home/oceanstar/1.avi";
        Mat frame;
        VideoCapture cap;
        cap.open(filename);
        if (!cap.isOpened()) {
        
            cerr << "ERROR! Unable to open camera\n";
            return -1;
        }
    
        if (!cap.read(frame)) {
        
            cerr << "ERROR! blank frame grabbed\n";
            return -1;
        }
    
        resize(frame, frame, Size(frame.cols / 2, frame.rows / 2), 0, 0, INTER_LINEAR_EXACT);
    
        Ptr<TrackerKCF> tracker = TrackerKCF::create();
        Rect2d rect = selectROI("frame", frame, false, false);
        //Rect2d rect(287, 23, 86, 320);
        tracker->init(frame, rect);
        cout << "starting tracking!" << endl;
    
        for (;;)
        {
        
            resize(frame, frame, Size(frame.cols / 2, frame.rows / 2), 0, 0, INTER_LINEAR_EXACT);
            tracker->update(frame, rect);
            rectangle(frame, rect, Scalar(0, 0, 255), 2, 1);
            imshow("Live", frame);
    
            cap.read(frame);
            if (frame.empty()) {
        
                cerr << "ERROR! blank frame grabbed\n";
                break;
            }
    
            if (waitKey(5) >= 0)
                break;
        }
        cap.release();
    
        return 0;
    }

基于模板的跟踪算法【效果很差】

#include <iostream>
    #include <opencv2/opencv.hpp>
    #include <opencv2/tracking.hpp>
    
    using namespace std;
    using namespace cv;
    
    Rect match(Mat &Image, Mat &templateImage, int method);
    
    int main(int argc, char *argv[])
    {
        
        VideoCapture cap("rtmp://*******");
        if (!cap.isOpened())
        {
        
            cout << "capture not Open!" << endl;
            return 0;
        }
    
        cap.set(CAP_PROP_FRAME_WIDTH, 600);
        cap.set(CAP_PROP_FRAME_HEIGHT, 600);
    
        Mat frame;
        cap.read(frame);
        imshow("Image", frame);
        Rect roi = selectROI("Image", frame, false, false);
        Mat tempImage = frame(roi);
        Mat grayTemp;
        cvtColor(tempImage, grayTemp, CV_BGR2GRAY);
        while (cap.read(frame))
        {
        
            Mat Image;
            cvtColor(frame, Image, CV_BGR2GRAY);
            Rect trackroi = match(Image, grayTemp, CV_TM_SQDIFF);
    
            rectangle(frame, trackroi, Scalar(0, 0, 255), 1, 8, 0);
            imshow("frame", frame);
    
            char c = waitKey(50);
            if (c == 27) break;
        }
    
        waitKey(0);
        return 0;
    }
    
    
    Rect match(Mat &Image, Mat &templateImage, int method)
    {
        
        int result_cols = Image.cols - templateImage.cols + 1;
        int result_rows = Image.rows - templateImage.rows + 1;
    
        Mat result = Mat(result_cols, result_rows, CV_32FC1);
    
        matchTemplate(Image, templateImage, result, method);
        normalize(result, result, 0, 1, NORM_MINMAX);
    
        double minVal, maxVal;
        Point minLoc, maxLoc, matchLoc;
        minMaxLoc(result, &minVal, &maxVal, &minLoc, &maxLoc, Mat());
    
        switch (method)
        {
        
            case CV_TM_SQDIFF:
            case CV_TM_SQDIFF_NORMED:
                matchLoc = minLoc;
                break;
            default:
                matchLoc = maxLoc;
                break;
        }
    
        rectangle(Image, Rect(matchLoc, Size(templateImage.cols, templateImage.rows)), Scalar(0, 0, 255), 2, 8, 0);
    
        return Rect(matchLoc, Size(templateImage.cols, templateImage.rows));
    }