src/LibFace.h

Go to the documentation of this file.

#ifndef _LIBFACE_H_
#define _LIBFACE_H_

#include "Eigenfaces.h"
#include "LibFaceConfig.h"
#include "LibFaceCore.h"

namespace libface {

enum Mode {
        ALL,
        DETECT,
        EIGEN
};

class FACEAPI LibFace {

public:

#ifdef WIN32
        LibFace(Mode type = ALL, const std::string& configDir = ".", const std::string& cascadeDir = std::string(OPENCVDIR)+"/data/haarcascades");
#else
        LibFace(Mode type = ALL, const std::string& configDir = ".", const std::string& cascadeDir = std::string(OPENCVDIR)+"/haarcascades");
#endif

        ~LibFace();

        // OpenCV compatibility methods

        int count() const;


        std::vector<Face>* detectFaces(const IplImage* img, const CvSize& originalSize);
        // API-agnostic methods

        std::vector<Face>* detectFaces(const std::string& filename, int scaleFactor=1);

        std::vector<Face>* detectFaces(const char* image, int width , int height , int step , int depth = IPL_DEPTH_8U, int channels = 1, int scaleFactor=1 );



        std::map<std::string, std::string> getConfig();

        double getDetectionAccuracy() const;

        int getRecommendedImageSizeForDetection(const CvSize& size = cvSize(0,0)) const;

        CvSize getRecommendedImageSizeForRecognition(const CvSize& size = cvSize(0,0)) const;

        int loadConfig(const std::string& configDir);

        int loadConfig(const std::map<std::string, std::string>& config);

        std::vector<std::pair<int, float> > recognise(const IplImage* img, std::vector<Face>* faces_p, int scaleFactor=1);


        std::vector<std::pair<int, float> > recognise(const std::string& filename, std::vector<Face>* faces_p, int scaleFactor=1);

        std::vector<std::pair<int, float> > recognise(const char* image, std::vector<Face>* faces_p, int width, int height, int step, int depth = IPL_DEPTH_8U, int channels = 1, int scaleFactor=1);

        std::vector<std::pair<int, float> > recognise( std::vector<Face>* faces, int scaleFactor =1);


        int saveConfig(const std::string& configDir);

        void setDetectionAccuracy(double value);


        int update(const std::string& filename, std::vector<Face>* faces=NULL, int scaleFactor=1);

        int update(const char* image, std::vector<Face>* faces, int width , int height , int step , int depth = IPL_DEPTH_8U, int channels = 1, int scaleFactor=1);

        int update(std::vector<Face>* faces=NULL, int scaleFactor=1);

        int update(const IplImage* img, std::vector<Face>* faces, int scaleFactor = 1);

private:

        class LibFacePriv;
        LibFacePriv* const d;
};

} // namespace libface

#endif /* _LIBFACE_H_ */