Meanshift跟踪移动目标

来源:互联网 发布:winxp哪个版本适合编程 编辑:程序博客网 时间:2024/05/23 02:04

利用Meanshift算法进行了移动目标跟踪,图片资源如下:

链接:https://yunpan.cn/cq9GfQDrsDL7r  密码:b135

红色方框为初始位置,绿色方框为跟踪到的位置。

 

#include <opencv2/opencv.hpp>using namespace cv;int histSize[1] = {256};float hranges[2] = {0, 255};const float* ranges[1] = {hranges};int channels[1] = {0};int main(){VideoCapture cap("E:/Datasets/img%4d.jpg");Rect rect(200, 115, 45, 45);Mat frame, frameHSV, imgROI, dstHist, backproj;cap >> frame;cvtColor(frame, frameHSV, COLOR_RGB2HSV);imgROI = frameHSV(rect);calcHist(&imgROI, 1, channels, Mat(), dstHist, 1, histSize, ranges);normalize(dstHist, dstHist, 0.0, 1.0, NORM_MINMAX);rectangle(frame, rect, Scalar(0, 0, 255));imshow("Frame", frame);waitKey(0);while (true){cap >> frame;if (frame.empty()) break;cvtColor(frame, frameHSV, COLOR_RGB2HSV);calcBackProject(&frameHSV, 1, channels, dstHist, backproj, ranges, 255.0);TermCriteria criteria(TermCriteria::MAX_ITER + TermCriteria::EPS, 1000, 0.001);meanShift(backproj, rect, criteria);imgROI = frameHSV(rect);calcHist(&imgROI, 1, channels, Mat(), dstHist, 1, histSize, ranges);normalize(dstHist, dstHist, 0.0, 1.0, NORM_MINMAX);rectangle(frame, rect, Scalar(0, 255, 0));imshow("Frame", frame);waitKey(30);}return 0;}


0 0