facerec.hpp

// This file is part of OpenCV project.
// It is subject to the license terms in the LICENSE file found in the top-level directory
// of this distribution and at http://opencv.org/license.html.
 
// Copyright (c) 2011,2012. Philipp Wagner <bytefish[at]gmx[dot]de>.
// Third party copyrights are property of their respective owners.
 
#ifndef __OPENCV_FACEREC_HPP__
#define __OPENCV_FACEREC_HPP__
 
#include "opencv2/face.hpp"
#include "opencv2/core.hpp"
 
namespace cv { namespace face {
 
 
// base for two classes
class CV_EXPORTS_W BasicFaceRecognizer : public FaceRecognizer
{
public:
    CV_WRAP int getNumComponents() const;
    CV_WRAP void setNumComponents(int val);
    CV_WRAP double getThreshold() const CV_OVERRIDE;
    CV_WRAP void setThreshold(double val) CV_OVERRIDE;
    CV_WRAP std::vector<cv::Mat> getProjections() const;
    CV_WRAP cv::Mat getLabels() const;
    CV_WRAP cv::Mat getEigenValues() const;
    CV_WRAP cv::Mat getEigenVectors() const;
    CV_WRAP cv::Mat getMean() const;
 
    virtual void read(const FileNode& fn) CV_OVERRIDE;
    virtual void write(FileStorage& fs) const CV_OVERRIDE;
    virtual bool empty() const CV_OVERRIDE;
 
    using FaceRecognizer::read;
    using FaceRecognizer::write;
 
protected:
    int _num_components;
    double _threshold;
    std::vector<Mat> _projections;
    Mat _labels;
    Mat _eigenvectors;
    Mat _eigenvalues;
    Mat _mean;
};
 
class CV_EXPORTS_W EigenFaceRecognizer : public BasicFaceRecognizer
{
public:
    CV_WRAP static Ptr<EigenFaceRecognizer> create(int num_components = 0, double threshold = DBL_MAX);
};
 
class CV_EXPORTS_W FisherFaceRecognizer : public BasicFaceRecognizer
{
public:
    CV_WRAP static Ptr<FisherFaceRecognizer> create(int num_components = 0, double threshold = DBL_MAX);
};
 
 
class CV_EXPORTS_W LBPHFaceRecognizer : public FaceRecognizer
{
public:
    CV_WRAP virtual int getGridX() const = 0;
    CV_WRAP virtual void setGridX(int val) = 0;
    CV_WRAP virtual int getGridY() const = 0;
    CV_WRAP virtual void setGridY(int val) = 0;
    CV_WRAP virtual int getRadius() const = 0;
    CV_WRAP virtual void setRadius(int val) = 0;
    CV_WRAP virtual int getNeighbors() const = 0;
    CV_WRAP virtual void setNeighbors(int val) = 0;
    CV_WRAP virtual double getThreshold() const CV_OVERRIDE = 0;
    CV_WRAP virtual void setThreshold(double val) CV_OVERRIDE = 0;
    CV_WRAP virtual std::vector<cv::Mat> getHistograms() const = 0;
    CV_WRAP virtual cv::Mat getLabels() const = 0;
 
    CV_WRAP static Ptr<LBPHFaceRecognizer> create(int radius=1, int neighbors=8, int grid_x=8, int grid_y=8, double threshold = DBL_MAX);
};
 
 
}} //namespace cv::face
 
#endif //__OPENCV_FACEREC_HPP__