Meanshift跟踪移动目标

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