基于头部跟踪初代码测试
#include "iostream" #include "queue" #include "Windows.h" #include <opencv2/ml/ml.hpp> #include "opencv2/opencv.hpp" #include "Windows.h" #include "opencv2/core/core.hpp" #include "opencv2/highgui/highgui.hpp" #include "opencv2/imgproc/imgproc.hpp" #include "opencv2/objdetect/objdetect.hpp" #include "opencv2/features2d/features2d.hpp" #include "opencv2/objdetect/objdetect.hpp" #include "opencv2/features2d/features2d.hpp" #include "opencv2/calib3d/calib3d.hpp" #include "opencv2/ml.hpp" #include "opencv/cv.h" #include "opencv/ml.h" #define MAX 30 using namespace cv; using namespace std; //画直方图用 int HistogramBins = 256; float HistogramRange1[2] = { 0, 255 }; float *HistogramRange[1] = { &HistogramRange1[0] }; /* * imagefile1: * imagefile2: * method: could be CV_COMP_CHISQR, CV_COMP_BHATTACHARYYA, CV_COMP_CORREL, CV_COMP_INTERSECT */ double CompareHist(IplImage *image1, IplImage *image2) { CvHistogram *Histogram1 = cvCreateHist(1, &HistogramBins, CV_HIST_ARRAY, HistogramRange); CvHistogram *Histogram2 = cvCreateHist(1, &HistogramBins, CV_HIST_ARRAY, HistogramRange); cvCalcHist(&image1, Histogram1); cvCalcHist(&image2, Histogram2); cvNormalizeHist(Histogram1, 1); cvNormalizeHist(Histogram2, 1); // CV_COMP_CHISQR,CV_COMP_BHATTACHARYYA这两种都可以用来做直方图的比较,值越小,说明图形越相似 return cvCompareHist(Histogram1, Histogram2, CV_COMP_CHISQR); //printf("CV_COMP_BHATTACHARYYA : %.4f\n", cvCompareHist(Histogram1, Histogram2, CV_COMP_BHATTACHARYYA)); // CV_COMP_CORREL, CV_COMP_INTERSECT这两种直方图的比较,值越大,说明图形越相似 //printf("CV_COMP_CORREL : %.4f\n", cvCompareHist(Histogram1, Histogram2, CV_COMP_CORREL)); //printf("CV_COMP_INTERSECT : %.4f\n", cvCompareHist(Histogram1, Histogram2, CV_COMP_INTERSECT)); //cvReleaseImage(&image1); //cvReleaseImage(&image2); cvReleaseHist(&Histogram1); cvReleaseHist(&Histogram2); } Mat Getmat(Mat mat, Rect rect) { return mat(rect); } int main() { VideoCapture video(0); if (!video.isOpened()) { return -1; } Mat img; Mat img1,img2; int num = 0; Rect rect(270, 130, 120, 160), rected[MAX]; //int diswidth[MAX] = { 1, 1, 0, -1, -1, -1, 0, 1, 2, 2, 0, -2, -2, -2, 0, 2}; //int dishigth[MAX] = { 0, 1, 1, 1, 0, -1, -1, -1, 0, 2, 2, 2, 0, -2, -2, -2}; int rang = 0; int diswidth[MAX]; int dishigth[MAX]; for (int i = -2; i <= 2; i++) { for (int j = -2; j <= 2; j++) { if (i == 0 && j == 0) continue; diswidth[rang] = i; dishigth[rang] = j; rang++; } } int mul = 2; double ans=0,minans; int id; IplImage frame1, frame2; IplImage *pframe1=NULL, *pframe2=NULL; IplImage *pgrayframe1 = NULL, *pgrayframe2 = NULL; /* cv::Mat img2; IplImage imgTmp = img2; IplImage *input = cvCloneImage(&imgTmp); */ int i; //get first video >> img; img1 = Getmat(img, rect); pframe1 = &(IplImage(img1)); pgrayframe1 = cvCreateImage(cvGetSize(pframe1), IPL_DEPTH_8U, 1); pgrayframe2 = cvCreateImage(cvGetSize(pframe1), IPL_DEPTH_8U, 1); cvCvtColor(pframe1, pgrayframe1, CV_BGR2GRAY); while (1) { video >> img; minans = 0x3f3f3f3f; id = 0; for (i = 0; i < rang; i++) { rected[i] = rect; rected[i].x += diswidth[i] * mul; rected[i].y += dishigth[i] * mul; rected[i].x = rected[i].x>0 ? rected[i].x : 0; rected[i].y = rected[i].y>0 ? rected[i].y : 0; rected[i].x = rected[i].x + rected[i].width <img.cols ? rected[i].x : img.cols-rected[i].width; rected[i].y = rected[i].y + rected[i].height<img.rows ? rected[i].y : img.rows-rected[i].height; img2 = Getmat(img, rected[i]); pframe2 = &(IplImage(img2)); cvCvtColor(pframe2, pgrayframe2, CV_BGR2GRAY); ans = CompareHist(pgrayframe1, pgrayframe2); if (ans < minans) { minans = ans; id = i; } } rect = rected[id]; img1 = Getmat(img, rect); //frame1 = img1; //pframe1 = cvCloneImage(&frame1); pframe1 = &(IplImage(img1)); cvCvtColor(pframe1, pgrayframe1, CV_BGR2GRAY); rectangle(img, rect, Scalar(0, 255, 0), 1); //imshow("vi", img1); //imshow("vi1", img2); //printf("%.7lf\n", ans); //img1 = img(rect); imshow("video", img); if (waitKey(25) > 0) { break; } } return 0; }