学习opencv 使用反向块投影搜寻图像中的物体的位置 cvCalcBackProjectPatch-CSDN博客

本文链接：https://blog.csdn.net/fdl19881/article/details/6726438

前一篇讲了使用cvCalcBackProject 寻找手的肤色地方

这一篇讲在一幅图像中寻找某个特定的小物体

步骤：

1.使用一幅只包含小物体的图片（图片大小和物体一样大即可，不需要向上一篇同样大），计算其二维直方图 (HSV空间中的 h(亮度）和 s （饱和度) 二维直方图 (还可以和图像的梯度和(或)梯度角度结合起来，查找纹理相似　和(或)　颜色相似

2.输入一幅待搜索的图片，转化为HSV空间 ,只取h 和 s 分量

3. 设置搜索块大小和物体大小相同（即和第1步中图片大小相同），使用cvCalcBackProjectPatch反向块投影和图片result

4.在result中取最大值所在位置，即可得到物体在输入图片中的位置了。 (或者设定一个阈值）

示例程序如下：

#include <iostream>
#include <cv.h>
#include <highgui.h>
#include <cxcore.h>

using namespace std;

void GetHSV(const IplImage *image,IplImage **h,IplImage **s,IplImage **v);

int main()
{
	IplImage *src = cvLoadImage("f:\\images\\bluecup.jpg");
	IplImage *h_src = NULL ,*s_src = NULL;
	GetHSV(src,&h_src,&s_src,NULL);
	IplImage *images[] = {h_src,s_src};
	CvHistogram *hist_src;
	{//计算二维直方图
		int dims = 2;
		int size[] = {30,32}; // 这个地方不要取的太大!
		//当取为size[] = {180,256}时E7200CPU会运行长达10几分钟的！
		float range_h[] = {0,180} //再用cvCvtColor转换时h已经归一化到180了
			,range_s[] = {0,256};
		float *ranges[] = {range_h,range_s};
		hist_src = cvCreateHist(dims,size,CV_HIST_ARRAY,ranges);
		cvCalcHist(images,hist_src);
		cvNormalizeHist(hist_src,1);
	}

	IplImage *dst = cvLoadImage("f:\\images\\adrian1.jpg");
	IplImage *h_dst = NULL,*s_dst = NULL;
	GetHSV(dst,&h_dst,&s_dst,NULL);
	images[0] = h_dst ,images[1] = s_dst; 

	CvSize patch_size = cvSize(src->width,src->height);
	IplImage *result = cvCreateImage(cvSize(h_dst->width - patch_size.width +1,h_dst->height - patch_size.height +1)
		,IPL_DEPTH_32F,1);//块搜索时处理边缘是直接舍去，故result的大小比dst小path_size大小
	//32F类型，取值为0~1最亮为1，可直接显示
	//CV_COMP_CORREL相关度，1时最匹配，0时最不匹配
	cvCalcBackProjectPatch(images,result,patch_size,hist_src,CV_COMP_CORREL,1);
	cvShowImage("result",result);
	
	//找出最大值位置，可得到此位置即为杯子所在位置
	CvPoint max_location;
	cvMinMaxLoc(result,NULL,NULL,NULL,&max_location,NULL);
	//加上边缘，得到在原始图像中的实际位置
	max_location.x += cvRound(patch_size.width/2);
	max_location.y += cvRound(patch_size.height/2);
	//在dst图像中用红色小圆点标出位置
	cvCircle(dst,max_location,3,CV_RGB(255,0,0),-1);

	cvShowImage("dst",dst);
	cvWaitKey();

	cvReleaseImage(&src);
	cvReleaseImage(&dst);
	cvReleaseImage(&h_src);
	cvReleaseImage(&h_dst);
	cvReleaseImage(&s_dst);
	cvReleaseImage(&s_src);
	cvReleaseHist(&hist_src);
	cvReleaseImage(&result);
	cvDestroyAllWindows();
}

void GetHSV(const IplImage *image , IplImage **h,IplImage **s,IplImage **v)
{
	IplImage *hsv = cvCreateImage(cvGetSize(image),8,3);
	cvCvtColor(image,hsv,CV_BGR2HSV);
	
	if((h != NULL) && (*h == NULL))
		*h = cvCreateImage(cvGetSize(image),8,1);
	if((s != NULL) && (*s == NULL))
		*s = cvCreateImage(cvGetSize(image),8,1);
	if((v != NULL) && (*v == NULL))
		*v = cvCreateImage(cvGetSize(image),8,1);


	cvSplit(hsv,*h,(s == NULL)?NULL:*s,(v==NULL)?NULL:*v,NULL);
	cvReleaseImage(&hsv);
}

第一步物体图片src刚好包含要搜索的物体：