使用Meanshift算法查找物体

看了很长时间的Meanshift，今天终于会用了。

借助于一本参考书《OpenCV2计算机视觉编程手册》，这本书帮了很大的忙。

 

 

#include <iostream>
#include <opencv2/opencv.hpp>

using namespace cv;

int main()
{
	Mat srcImg1 = imread("1.jpg");
	Mat srcImg2 = imread("2.jpg");

	Mat srcHSV1, srcHSV2;
	cvtColor(srcImg1, srcHSV1, COLOR_RGB2HSV);
	cvtColor(srcImg2, srcHSV2, COLOR_RGB2HSV);

	Rect rect(130, 120, 55, 55);
	Mat imgROI = srcHSV1(rect);
	rectangle(srcImg1, rect, Scalar(0, 0, 255));
	rectangle(srcImg2, rect, Scalar(0, 0, 255));

	imshow("srcImg1", srcImg1);
	imwrite("srcImg1.png", srcImg1);

	int histSize[1] = {256};
	float hranges[2] = {0, 255};
	const float* ranges[1] = {hranges};
	int channels[1] = {0};

	Mat dstHist;
	calcHist(&imgROI, 1, channels, Mat(), dstHist, 1, histSize, ranges);
	normalize(dstHist, dstHist, 0.0, 1.0, NORM_MINMAX);

	Mat backprojection;	
	calcBackProject(&srcHSV2, 1, channels, dstHist, backprojection, ranges, 255.0);
	imshow("backprojection", backprojection);
	imwrite("backproj.png", backprojection);

	TermCriteria criteria(TermCriteria::MAX_ITER + TermCriteria::EPS, 10, 0.01);
	meanShift(backprojection, rect, criteria);//Meanshift会自动更新这个rect
	rectangle(srcImg2, rect, Scalar(0, 255, 0));
	imshow("srcImg2", srcImg2);
	imwrite("srcImg2.png", srcImg2);

	waitKey(0);

	return 0;
}