pans

2017-01-03 22d3d8fccb06812fc20e0cfcf96cb9c241fd9e52

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

 RtspFace/demo/src/faceAPI.cpp

@@ -1,5 +1,4 @@
#include "faceAPI.h"

#include "faceDB.h"



faceDB *fdb;

faceAPI::faceAPI()

@@ -16,6 +15,9 @@


}



//@brief 搜索功能

//@param 图片

//@return 人脸id

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

{

    if (!image.data) {

@@ -32,12 +34,18 @@
    return p_img_search(p_f);

}



//@brief 搜索功能

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

//@return 人脸id

int faceAPI::do_reasch(char* s_feature)

{

    get_feature(s_feature,p_f);

    return p_img_search(p_f);

}



//@brief 注册功能，未完成

//@param 图片

//@return 人脸id

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

{

    p_f = fdb->extract_feature(image);

@@ -47,6 +55,9 @@
}





//@brief 在数据库中搜索人脸id

//@param 特征值

//@return 人脸id

int faceAPI::p_img_search(cv_feature_t *p_feature)

{

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

@@ -56,7 +67,8 @@


//@brief 特征值转换为字符串

//@param 特征值

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

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

//@return 状态量

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

{



@@ -65,7 +77,8 @@


//@brief 字符串转换为特征值

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

//@return 特征值

//@param 特征值

//@return 状态量

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

{



@@ -75,3 +88,157 @@
    } else

        return -1;

}


//faceDB实现部分

#include <iostream>



//"./out1.db"

char *db_path = "out.db";



faceDB::faceDB()

{

    // 创建人脸数据库句柄

    cv_verify_create_db(&handle_db);

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

    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");

}



faceDB::~faceDB() {}



/*

faceDB* faceDB::GetInstance(){

    if (db == NULL)

    {

        db = new faceDB();

    }



    return db;

}*/



//@brief 数据库添加数据

//@param 特征值

//@return 人脸id

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);

    }

    cv_verify_release_feature(p_feature);

    return idx;

}





//@brief 保存数据库

//@return 状态量

bool faceDB::db_save()

{

    cv_result = cv_verify_save_db(handle_db, db_path);

    if (cv_result != CV_OK) {

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

        return false;

    } else {

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

    }



    return true;

}



//@brief 加载数据库

//@return 状态量

bool faceDB::db_load()

{

    if (handle_db != NULL) {

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

    } else {

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

    }

    cv_result = cv_verify_load_db(handle_db, db_path);

    if (cv_result != CV_OK) {

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

        return false;

    } else {

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

    }



    return true;

}



//@brief 搜索数据库

//@param 人脸特征值

//@return 人脸id

int faceDB::search_db(cv_feature_t *p_feature)

{

    int indx=-1;

    //查询前10条

    int top_k = 3;

    int *top_idxs = new int[top_k];

    float *top_scores = new float[top_k];

    unsigned int result_length = 0;

    cv_result = cv_verify_search_face(handle_verify, handle_db,

                                      p_feature, top_k,

                                      top_idxs, top_scores, &result_length);





    if (cv_result == CV_OK) {

        //输出检查结果

        for (unsigned int t = 0; t < result_length; t++) {

            // const cv_feature_t item = result[t].item;

            fprintf(stderr, "%d\t", top_idxs[t]);

            fprintf(stderr, "%0.2f\n", top_scores[t]);

        }

    } else {

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

    }

    if (top_scores[0] != 0) {

        indx=top_idxs[0];

    }

    if (top_idxs) {

        delete[]top_idxs;

    }

    if (top_scores) {

        delete[]top_scores;

    }

    cv_verify_release_feature(p_feature);

    return indx;

}



//@brief 提取特征值

//@param 图片

//@return 特征值

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

{

    if (handle_verify != NULL) {

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

    } else {

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

    }



    cv_feature_t *p_feature = NULL;

    cv_face_t *p_face = NULL;

    int face_count = 0;

    cv_result_t st_result = CV_OK;

    st_result = cv_face_detect(handle_detect, image_color.data, CV_PIX_FMT_BGR888,

                               image_color.cols, image_color.rows, image_color.step,

                               CV_FACE_UP, &p_face, &face_count);

    if (face_count >= 1) {

        st_result = cv_verify_get_feature(handle_verify,

                                          (unsigned char *)image_color.data, CV_PIX_FMT_BGR888,

                                          image_color.cols, image_color.rows, image_color.step,

                                          p_face, &p_feature, NULL);

        if (st_result != CV_OK) {

            fprintf(stderr, "cv_verify_get_feature failed, error code %d\n", st_result);

        }

    } else {

        fprintf(stderr, "can't find face in ");

    }

    // release the memory of face

    cv_face_release_detector_result(p_face, face_count);

    return p_feature;

}

 RtspFace/demo/src/faceAPI.h

@@ -5,6 +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

{

public:

@@ -28,4 +35,33 @@
    cv_feature_t* p_f;

};





class faceDB

{

public:

    //static faceDB* GetInstance();



    //搜索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 & );



    //加载sdk数据库

    bool db_load();

};



#endif