faceAPI整合

pans

2017-01-11 307706a63521650ca1cc7eebff0a931b432539c3

faceAPI整合

git-svn-id: http://192.168.1.226/svn/proxy@90 454eff88-639b-444f-9e54-f578c98de674

 RtspFace/SensetimeFaceAPIWrapper/src/faceAPI.cpp

@@ -1,16 +1,17 @@
#include "faceAPI.h"

#include "FaceAPI.h"

#include <iostream>



faceDB *fdb;

faceAPI::faceAPI()

FaceDB *fdb;

FaceAPI::FaceAPI()

{

    //创建静态人脸识别句柄，同时创建状态标量

    cv_result_t cv_result = 0;



    //faceDB* f_db = faceDB.GetInstance();

    fdb=new faceDB();

    //FaceDB* f_db = FaceDB.GetInstance();

    fdb=new FaceDB();

}



faceAPI::~faceAPI()

FaceAPI::~FaceAPI()

{



}

@@ -18,7 +19,7 @@
//@brief 搜索功能

//@param 图片

//@return 人脸id

int faceAPI::do_reasch(cv::Mat image)

int FaceAPI::do_reasch(cv::Mat image)

{

    if (!image.data) {

        fprintf(stderr, "fail to read img\n");

@@ -37,7 +38,7 @@
//@brief 搜索功能

//@param 字符串格式的特征值

//@return 人脸id

int faceAPI::do_reasch(char* s_feature)

int FaceAPI::do_reasch(char* s_feature)

{

    get_feature(s_feature,p_f);

    return p_img_search(p_f);

@@ -46,7 +47,7 @@
//@brief 注册功能

//@param 图片

//@return 人脸id

int faceAPI::do_register(cv::Mat image)

int FaceAPI::do_register(cv::Mat image)

{

    p_f = fdb->extract_feature(image);

    int indx = fdb->db_add(p_f);

@@ -58,7 +59,7 @@
//@brief 在数据库中搜索人脸id

//@param 特征值

//@return 人脸id

int faceAPI::p_img_search(cv_feature_t *p_feature)

int FaceAPI::p_img_search(cv_feature_t *p_feature)

{

    std::cout<<"p img search"<<std::endl;



@@ -69,7 +70,7 @@
//@param 特征值

//@param 字符串格式的特征值

//@return 状态量

int faceAPI::get_char(cv_feature_t *p_feature,char* feature_str)

int FaceAPI::get_char(cv_feature_t *p_feature,char* feature_str)

{



    return cv_verify_serialize_feature(p_feature,feature_str);

@@ -79,7 +80,7 @@
//@param 字符串格式的特征值

//@param 特征值

//@return 状态量

int faceAPI::get_feature(char *feature_str,cv_feature_t *p_feature)

int FaceAPI::get_feature(char *feature_str,cv_feature_t *p_feature)

{



    p_feature = cv_verify_deserialize_feature(feature_str);

@@ -90,45 +91,65 @@
}



//faceDB实现部分

#include <iostream>



//"./out1.db"

char *db_path = "out.db";

//char *db_path = "out.db";



faceDB::faceDB()

FaceDB::FaceDB()

{

    // 创建人脸数据库句柄

    cv_verify_create_db(&handle_db);

    fprintf(stderr, "create cv_verify_create_db\n");

	
}



FaceDB::~FaceDB() {}



//初始化设置

bool FaceDB::init(char* db_path_t)

{

	
    db_path = db_path_t;

	
    //加载一次可多次使用。

    cv_verify_create_handle(&handle_verify, "../models/verify.model");

    fprintf(stderr, "create cv_verify_create_handle\n");

    cv_face_create_detector(&handle_detect, NULL, CV_DETECT_ENABLE_ALIGN_21);

    fprintf(stderr, "create cv_face_create_detector\n");

    db_load();

    fprintf(stderr, "db load\n");

	
	
    // 创建人脸数据库句柄，在db_save()中，数据库保存成功，会释放该句柄

    cv_verify_create_db(&handle_db);

    fprintf(stderr, "create cv_verify_create_db\n");

	
	
    if(db_load())

    {

        fprintf(stderr, "db load\n");

        return true;

    }

    fprintf(stderr, "db load falied\n");

    return false;

}



faceDB::~faceDB() {}



/*

faceDB* faceDB::GetInstance(){

    if (db == NULL)

//调用数据库保存

bool FaceDB::finally()

{

    if(db_save())

    {

        db = new faceDB();

        return true;

    }



    return db;

}*/



	
    return false;

	
}

	
//@brief 数据库添加数据

//@param 特征值

//@return 人脸id

int faceDB::db_add(cv_feature_t* p_feature)

//@return 人脸id，添加失败返回-1

int FaceDB::db_add(cv_feature_t* p_feature)

{

    int idx;

    cv_result_t cv_result = cv_verify_add_face(handle_db, p_feature, &idx);

    if (cv_result != CV_OK) {

        fprintf(stderr, "cv_verify_add_face failed, error code %d\n", cv_result);

        return -1；

    }

    cv_verify_release_feature(p_feature);

    return idx;

@@ -137,7 +158,7 @@


//@brief 保存数据库

//@return 状态量

bool faceDB::db_save()

bool FaceDB::db_save()

{

    cv_result = cv_verify_save_db(handle_db, db_path);

    if (cv_result != CV_OK) {

@@ -146,13 +167,13 @@
    } else {

        fprintf(stderr, "save done!\n");

    }



    cv_verify_destroy_db(handle_db);

    return true;

}



//@brief 加载数据库

//@return 状态量

bool faceDB::db_load()

bool FaceDB::db_load()

{

    if (handle_db != NULL) {

        fprintf(stderr, "handle_db is not null!%s\n",db_path);

@@ -173,7 +194,7 @@
//@brief 搜索数据库

//@param 人脸特征值

//@return 人脸id

int faceDB::search_db(cv_feature_t *p_feature)

int FaceDB::search_db(cv_feature_t *p_feature)

{

    int indx=-1;

    //查询前10条

@@ -215,7 +236,7 @@
//@brief 提取特征值

//@param 图片

//@return 特征值

cv_feature_t* faceDB::extract_feature(cv::Mat image_color)

cv_feature_t* FaceDB::extract_feature(cv::Mat image_color)

{

    if (handle_verify != NULL) {

        fprintf(stderr, "fdb.handle_verify is not null!\n");

@@ -223,7 +244,9 @@
        fprintf(stderr, "fdb.handle_verify is null!\n");

    }



    //该变量内存会在db_add()中，添加成功后释放。

    cv_feature_t *p_feature = NULL;

	
    cv_face_t *p_face = NULL;

    int face_count = 0;

    cv_result_t st_result = CV_OK;


 RtspFace/SensetimeFaceAPIWrapper/src/faceAPI.h

@@ -5,18 +5,13 @@
#include <cv_face.h>

#include "time_helper.h"



//人脸验证

static cv_handle_t handle_verify;

//数据库操作

static cv_handle_t handle_db;

//静态人脸识别

static cv_handle_t handle_detect;



class faceAPI



class FaceAPI

{

public:

    faceAPI();

    ~faceAPI();

    FaceAPI();

    ~FaceAPI();

    //查询  前台调用

    int do_reasch(char* s_feature);

    int do_reasch(cv::Mat image);

@@ -38,30 +33,38 @@
};





class faceDB

class FaceDB

{

public:

    //static faceDB* GetInstance();



    FaceDB();

    virtual ~FaceDB();

	
    //人脸验证

    cv_handle_t handle_verify;

    //数据库操作

    cv_handle_t handle_db;

    //静态人脸识别

    cv_handle_t handle_detect;



    //初始化

    bool init(char* db_path);

    //保存数据库

    bool finally();

	
    //搜索SDK数据库

    int search_db(cv_feature_t* p_feature );

    //添加记录

    int db_add(cv_feature_t* p_feature);

    //保存数据库

    bool db_save();

    int set_dbpath(char* db_path);



    faceDB();

    virtual ~faceDB();



    //提取特征值

    cv_feature_t *extract_feature(cv::Mat image_color);



private:



    cv_result_t cv_result;

    //static faceDB* db;

    //faceDB(const faceDB & );



    char* db_path;

	
    //加载sdk数据库

    bool db_load();

};