黑马程序员技术交流社区

标题: 【上海校区】稀疏光流跟踪 [打印本页]

---

作者: 梦缠绕的时候    时间: 2019-2-26 10:24
标题: 【上海校区】稀疏光流跟踪

/ 光流对象的跟踪.cpp: 定义控制台应用程序的入口点。
//分为稀疏光流 KLT  和稠密光流HF
//本课程用KLT

#include<opencv2/opencv.hpp>
#include<iostream>
using namespace cv;
using namespace std;

Mat frame, gray;
Mat pre_frame, pre_gray;
vector<Point2f>features;//Tomas角点检测特征点
vector<Point2f>inpoints;//初始化特征点数据
vector<Point2f>fpts[2];//保持当前帧和前一帧特征点位置

vector<uchar> status;//特征点各跟踪成功标志位
vector<float> errors;// 跟踪时候区域误差和
void detectcorner(Mat &inframe,Mat &ingray);//角点检测
void drawcorner(Mat &inframe);//角点显示
void track();//角点跟踪
void drawtracklines();
int main(int argc, char ** argv)
{
        VideoCapture capture;
        capture.open(0);
        if (!capture.isOpened())
        {
                printf("can not open ......\n");
                return -1;
        }
        namedWindow("output", WINDOW_AUTOSIZE);

        while (capture.read(frame))
        {
                cvtColor(frame, gray, CV_BGR2GRAY);//光流跟踪需要灰度图像
                if (fpts[0].size() < 40) //跟踪40个特征点，如果跟踪的时候损失了一些特征点，重新检测，追加
                {
                        detectcorner(frame, gray);//调用Tomas角点检测
                        fpts[0].insert(fpts[0].end(), features.begin(), features.end());// 追加带跟踪的特征点
                        inpoints.insert(inpoints.end(), features.begin(), features.end());
                }
                else
                {
                        cout << "track.......\n";// 表示特征点没有损失，一直在跟踪

                }
                if (pre_gray.empty())
                {
                        gray.copyTo(pre_gray); // 保存前一帧，第一帧过完就不保存了

                }
                track();//角点跟踪，KLT
                drawcorner(frame);//角点显示


                //更新前一帧数据
                gray.copyTo(pre_gray);
                frame.copyTo(pre_frame);

                imshow("output", frame);
                waitKey(1);
        }
        capture.release();//释放内存
        waitKey(0);
        return 0;
}

//角点检测
void detectcorner(Mat &inframe, Mat &ingray)
{
        double maxCorners = 5000;//检测到的角点的数量的最大值
        double  qualilevel = 0.01;//检测到的角点的质量等级，角点特征值小于qualityLevel*最大特征值的点将被舍弃
        double mindistant = 10;//两个角点间最小间距
        double bsize = 3;//计算协方差矩阵时窗口大小
        goodFeaturesToTrack(ingray, features, maxCorners, qualilevel, mindistant, Mat(), bsize, false, 0.04);//Tomas角点检测
        //输入必须为灰度图像，features保存检测到的角点坐标。Mat()检测整幅图像。true使用Harris角点检测，为false，则使用Shi-Tomasi算子
        //留给Harris角点检测算子用的中间参数，一般取经验值0.04~0.06。第八个参数为false时，该参数不起作用；
        //稀疏光流也用到了相邻三层金字塔，只是使用默认值没有参数显示
        cout << "detect" << features.size()<<endl;
}
//角点显示
void drawcorner(Mat &inframe) {
        for (int i = 0; i < fpts[0].size(); i++)
        {
                circle(inframe, fpts[0][i], 2,Scalar(0, 0, 255));
        }
}

void track()
{
        //输入前衣服图像和当前图像。输入前一副图像特性点，如果在后一副图像追踪到了前一副图像的特征点，则fpts[1]保存该特征点坐标
        //status如果相应位置的流特征被发现，向量的每个元素被设置为1，否则，被置为0.
        calcOpticalFlowPyrLK(pre_gray, gray, fpts[0], fpts[1], status, errors);//KLT,稀疏光流
        int k = 0;// 保存跟踪到的特征点数
        //特征点过滤
        for (int i = 0; i < fpts[1].size(); i++)
        {
                double dist = abs(fpts[0][i].x - fpts[1][i].x) + abs(fpts[0][i].y - fpts[1][i].y);
                if (dist > 2 && status[i])// 跟踪到的特征点，且距离移动了2以上的
                {
                        inpoints[k] = inpoints[i];// 将跟踪到的移动了的特征点在vector中连续起来，剔掉损失的和静止不动的特征点(这些跟踪点在前面帧中)
                        fpts[1][k++] = fpts[1][i];// 同上 (只是这些跟踪点在当前帧中)
                }
        }
        //保存特征点并绘制跟踪轨迹
        inpoints.resize(k);//重新设置容器大小
        fpts[1].resize(k);//重新设置容器大小
        drawtracklines();//画线
        std::swap(fpts[1], fpts[0]);// 交换，将此帧跟踪到的特征点作为下一帧的待跟踪点
}

void drawtracklines()
{
        for (int t = 0; t < fpts[1].size(); t++)
        {
                line(frame, inpoints[t], fpts[1][t], Scalar(0, 255, 0), 2, 8, 0);//画线
                circle(frame, fpts[1][t], 2, Scalar(0, 0, 255));
        }
}

---

作者: 不二晨    时间: 2019-2-26 15:36
奈斯，感谢分享

---

欢迎光临 黑马程序员技术交流社区 (http://bbs.itheima.com/)

黑马程序员IT技术论坛 X3.2