【opencv】均值漂移函数

最新推荐文章于 2022-01-03 19:34:51 发布

原创最新推荐文章于 2022-01-03 19:34:51 发布 · 954 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#均值漂移函数 #opencv

opencv 专栏收录该内容

49 篇文章

订阅专栏

本文介绍了一种基于均值漂移的目标跟踪方法。通过计算目标区域的颜色直方图，并利用该直方图进行反向投影，从而实现目标在连续帧间的定位与跟踪。文章详细阐述了关键步骤，包括直方图计算、反向投影及均值漂移迭代。

均值漂移前要得到目标的Mat（choseAim）以及Rect（rect）

首先计算第一次的dstHist

然后将dstHist，rect，image（待处理图原图，下一帧）扔到均值漂移函数中

之后循环即可

//均值漂移参数
Mat choseAim;
Mat targetImageHSV;
int histSize=200;  
float histR[]={0,255};  
const float *histRange=histR;  
int channels[]={0,1}; 
Mat dstHist;
//计算dstHist，以便均值漂移
cvtColor(choseAim, targetImageHSV, CV_RGB2HSV);
calcHist(&targetImageHSV, 2, channels, Mat(), dstHist, 1, &histSize, &histRange, true, false);  		
normalize(dstHist, dstHist, 0, 255, CV_MINMAX);

//************************************
// Description: 均值漂移函数
// Method:    meanShiftTracking
// FullName:  meanShiftTracking
// Access:    public 
// Parameter: Mat image 待检测图像
// Parameter: Rect & rect 上一帧目标位置
// Parameter: Mat dstHist 上一帧目标hist
// Returns:   void
// Author:    小白
// Date:      2017/07/03
//************************************
void meanShiftTracking(Mat image, Rect& rect, Mat dstHist)
{
    Mat imageHSV;
    Mat calcBackImage;
    cvtColor(image, imageHSV, CV_RGB2HSV);
    calcBackProject(&imageHSV, 2, channels, dstHist, calcBackImage, &histRange);  //反向投影
    TermCriteria criteria(TermCriteria::MAX_ITER + TermCriteria::EPS, 1000, 0.001);  
    meanShift(calcBackImage, rect, criteria);   
    Mat imageROI=imageHSV(rect);   //更新模板			
    targetImageHSV=imageHSV(rect);
    calcHist(&imageROI, 2, channels, Mat(), dstHist, 1, &histSize, &histRange);  
    normalize(dstHist, dstHist, 0.0, 1.0, NORM_MINMAX);   //归一化
    rectangle(image, rect, Scalar(0, 0, 255), 3);  	//目标绘制	
}