基于头部跟踪初代码测试

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