Exemplo n.º 1
0
void HandPairCollection::makePairs( const cv::PCA &pca,
                                    const ContourSet &singles ) {
    int size = singles.size();
    hands.reserve( size * size + 2 );
    centers.reserve( size * size + 2 );
    for( int dom = 0; dom < size; ++dom ) {
        for( int weak = 0; weak < size; ++weak ) {
            Projection proj;
            CenterPair centerp;
            if( weak != dom ) {
                proj = pca.project( flattenHistogram(
                                        generateHandHistogram( singles[dom], &singles[weak]) ) );
                centerp = std::make_pair( ::center( singles[dom] ),
                                          new CenterPoint( ::center( singles[weak] ) ) );
            }
            else {
                proj = pca.project(
                           flattenHistogram( generateHandHistogram( singles[dom], (Contour *)0 ) ));
                centerp = std::make_pair( ::center( singles[dom] ), (CenterPoint *)0 );
            }
            hands.push_back( proj );
            centers.push_back( centerp );
        }
    }
}
Exemplo n.º 2
0
void testSVM(vector<string>& pPositiveFiles, vector<string>& pNegativeFiles, Descriptor_Hog& pDescriptor, CvSVM& pSvm, cv::PCA pPca, float pPortion = 1.0)
{
    float portion;
    if (pPortion < 1.f)
        portion = pPortion;
    else
        portion = 0.f;

    // We compute the total detection time (descriptor creation + prediction)
    unsigned long long totalTime = 0;
    unsigned long long chronoTime;
    auto chronoStart = std::chrono::high_resolution_clock::now();

    /***/
    if (!gTable)
        cout << "Testing positive files... " << flush;

    int positive = 0;
    int negative = 0;
    int portionOfPositiveFiles = (int)((float)pPositiveFiles.size() * portion);
    for (int i = portionOfPositiveFiles; i < pPositiveFiles.size(); ++i)
    {
        cv::Mat image = cv::imread(pPositiveFiles[i]);

        chronoStart = std::chrono::high_resolution_clock::now();
        chronoTime = chrono::duration_cast<chrono::microseconds>(chronoStart.time_since_epoch()).count();

        pDescriptor.setImage(image);

        vector<float> description = pDescriptor.getDescriptor(_roiPosition);
        cv::Mat descriptionMat(1, (int)description.size(), CV_32FC1, &description[0]);

        // If we used PCA for training
        if (gPca < 1.0)
            descriptionMat = pPca.project(descriptionMat);

        float value = pSvm.predict(descriptionMat, true);

        if (gVerbose)
            cout << pPositiveFiles[i] << " -> " << value << endl;

        if (value < -gMargin)
            positive++;

        totalTime += timeSince(chronoTime);
    }

    /***/
    if (!gTable)
        cout << "Testing negative files" << endl;

    int totalNegatives = 0;
    int portionOfNegativeFiles = (int)((float)pNegativeFiles.size() * portion);
    for (int i = portionOfNegativeFiles; i < pNegativeFiles.size(); ++i)
    {
        cv::Mat image = cv::imread(pNegativeFiles[i]);
        chronoStart = std::chrono::high_resolution_clock::now();
        chronoTime = chrono::duration_cast<chrono::microseconds>(chronoStart.time_since_epoch()).count();
        pDescriptor.setImage(image);

        for (int x = 0; x < image.cols - _roiSize.x; x += image.cols/5)
        {
            for (int y = 0; y < image.rows - _roiSize.y; y += image.rows/5)
            {
                vector<float> description = pDescriptor.getDescriptor(_roiPosition + cv::Point_<int>(x, y));
                cv::Mat descriptionMat(1, (int)description.size(), CV_32FC1, &description[0]);
                if (description.size() == 0)
                    continue;

                // If we used PCA for training
                if (gPca < 1.0)
                    descriptionMat = pPca.project(descriptionMat);

                float value = pSvm.predict(descriptionMat, true);

                if (gVerbose)
                    cout << pNegativeFiles[i] << " -> " << value << endl;

                totalNegatives++;
                if (value > -gMargin)
                    negative++;
            }
        }

        totalTime += timeSince(chronoTime);
    }

    if (!gTable)
    {
        cout << "Positive: " << positive << " / " << pPositiveFiles.size() - portionOfPositiveFiles << endl;
        cout << "Negative: " << negative << " / " << totalNegatives << endl;
        cout << "Time per prediction (us): " << totalTime / (pPositiveFiles.size() - portionOfPositiveFiles + totalNegatives) << endl;
    }
    else
    {
        cout << gIterations << " ";
        cout << gEpsilon << " ";
        cout << gCPenalty << " ";
        cout << gOutput << " ";
        cout << gPosition << " ";
        cout << gCellSize << " ";
        cout << gBlockSize << " ";
        cout << gRoiSize << " ";
        cout << gBins << " ";
        cout << gSigma << " ";
        cout << gPca << " ";
        cout << totalTime / (pPositiveFiles.size() - portionOfPositiveFiles + totalNegatives) << " ";
        cout << (float)positive / ((float)pPositiveFiles.size() - (float)portionOfPositiveFiles) << " " << (float)negative/(float)totalNegatives << endl;
    }
}
Exemplo n.º 3
0
void trainSVM(vector<string>& pPositiveFiles, vector<string>& pNegativeFiles, Descriptor_Hog& pDescriptor, CvSVM& pSvm, cv::PCA& pPca, float pPortion = 1.0)
{
    int result;

    if (!gTable)
    {
        cout << "Training parameters: " << endl;
        if (_svmCriteria == CV_TERMCRIT_ITER)
            cout << "   iterations = " << gIterations << endl;
        if (_svmCriteria == CV_TERMCRIT_EPS)
            cout << "   epsilon = " << gEpsilon << endl;
        cout << "   C penalty = " << gCPenalty << endl;
        cout << "   output file = " << gOutput << endl;
        cout << "   ROI position = " << gPosition << endl;
        cout << "   cell size = " << gCellSize << endl;
        cout << "   block size = " << gBlockSize << endl;
        cout << "   roi size = " << gRoiSize << endl;
        cout << "   bins per cell = " << gBins << endl;
        cout << "   sigma = " << gSigma << endl;
        cout << "   pca = " << gPca << endl;
    }
    
    // Set up training data
    if (!gVerbose && !gTable)
        cout << "Loading training data... " << flush;

    int nbrImg = 0;
    vector<float> labels;
    vector<float> trainingData;
    int portionOfPositiveFiles = (int)((float)pPositiveFiles.size() * pPortion);
    for (int i = 0; i < portionOfPositiveFiles; ++i)
    {
        if (gVerbose)
            cout << "Analysing " << pPositiveFiles[i] << endl;

        cv::Mat image = cv::imread(pPositiveFiles[i]);
        pDescriptor.setImage(image);
        vector<float> description = pDescriptor.getDescriptor(_roiPosition);
        if (description.size() == 0)
            continue;

        labels.push_back(1.f);
        for (int j = 0; j < description.size(); ++j)
            trainingData.push_back(description[j]);

        nbrImg++;
    }
    if (!gVerbose && !gTable)
        cout << "Positive data loaded... " << flush;

    int portionOfNegativeFiles = (int)((float)pNegativeFiles.size() * pPortion);
    for (int i = 0; i < portionOfNegativeFiles; ++i)
    {
        if (gVerbose)
            cout << "Analysing " << pNegativeFiles[i] << endl;

        cv::Mat image = cv::imread(pNegativeFiles[i]);
        pDescriptor.setImage(image);
        for (int x = 0; x < image.cols - _roiSize.x; x += image.cols/5)
        {
            for (int y = 0; y < image.rows - _roiSize.y; y += image.rows/5)
            {
                vector<float> description = pDescriptor.getDescriptor(_roiPosition + cv::Point_<int>(x, y));
                if (description.size() == 0)
                    continue;

                labels.push_back(-1.f);
                for (int j = 0; j < description.size(); ++j)
                    trainingData.push_back(description[j]);

                nbrImg++;
            }
        }
    }
    if (!gVerbose && !gTable)
        cout << "Negative data loaded" << endl;

    // Applying the PCA and reformating input data
    if (gPca < 1.0)
    {
        cout << "Applying PCA before SVM train... ";
        cout.flush();

        int vectorSize = trainingData.size() / nbrImg;
        cv::Mat trainingPCA = cv::Mat::zeros(nbrImg, vectorSize, CV_32F);
        for (int i = 0; i < nbrImg; ++i)
        {
            for (int j = 0; j < vectorSize; ++j)
            {
                trainingPCA.at<float>(i, j) = trainingData[i*vectorSize + j];
            }
        }
        int keptComponents = (int)((float)vectorSize * gPca);
        pPca(trainingPCA, cv::noArray(), CV_PCA_DATA_AS_ROW, keptComponents);

        // Reproject all vectors
        cv::Mat projected;
        pPca.project(trainingPCA, projected);

        // Store the result as the training data
        trainingData.clear();
        for (int i = 0; i < projected.rows; ++i)
            for (int j = 0; j < projected.cols; ++j)
            {
                trainingData.push_back(projected.at<float>(i, j));
            }

        cout << "Kept " << keptComponents << " components" << endl;

        // Save the resulting space
        cv::FileStorage file("pca.xml", cv::FileStorage::WRITE);
        file << "eigenVectors" << pPca.eigenvectors;
        file << "mean" << pPca.mean;
    }
    
    if (!gTable)
        cout << "Training the SVM classifier..." << endl;
    // Set up the SVM train parameter
    CvSVMParams svmParams;
    svmParams.svm_type = CvSVM::C_SVC;
    svmParams.C = gCPenalty;
    svmParams.kernel_type = CvSVM::LINEAR;
    svmParams.term_crit = cvTermCriteria(_svmCriteria, gIterations, gEpsilon);

    // Train the SVM
    cv::Mat labelsMat((int)labels.size(), 1, CV_32FC1, &labels[0]);
    cv::Mat trainingMat(nbrImg, (int)trainingData.size() / nbrImg, CV_32FC1, &trainingData[0]);
    result = pSvm.train(trainingMat, labelsMat, cv::Mat(), cv::Mat(), svmParams);

    pSvm.save((const char*)gOutput);
}