pans

2017-01-04 7bcd2613bb116eff33d6e2358bd8c737bacc3cc1

 RtspFace/demo/src/faceAPI.cpp

@@ -1,70 +1,244 @@
#include "faceAPI.h"

#include "faceDB.h"



faceDB *fdb;

faceAPI::faceAPI(){

faceAPI::faceAPI()

{

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

   cv_result_t cv_result = 0;

	


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

   fdb=new faceDB();

}



faceAPI::~faceAPI(){

faceAPI::~faceAPI()

{



}



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

//@brief 搜索功能

//@param 图片

//@return 人脸id

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

{

   if (!image.data) {

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

      return -1;

   }else

   } else

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

   p_f = fdb->extract_feature(image);

   fprintf(stderr, "fdb->extract_feature(image)\n");

   if (!p_f)

   {

   if (!p_f) {

      fprintf(stderr, "p_f is null\n");

      return -2;

   }

   return p_img_search(p_f); 
   return p_img_search(p_f);

}



int faceAPI::do_reasch(char* s_feature){

//@brief 搜索功能

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

//@return 人脸id

int faceAPI::do_reasch(char* s_feature)

{

   get_feature(s_feature,p_f);

   return p_img_search(p_f);

}



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

//@brief 注册功能

//@param 图片

//@return 人脸id

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

{

   p_f = fdb->extract_feature(image);

   //db add

   int indx = 1;

   return indx;

   int indx = fdb->db_add(p_f);

   if(fdb->db_save())

      return indx;

   return -2;

}





int faceAPI::p_img_search(cv_feature_t *p_feature){

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

//@param 特征值

//@return 人脸id

int faceAPI::p_img_search(cv_feature_t *p_feature)

{

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

	


   return fdb->search_db(p_feature);

}



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

//@param 特征值

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

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

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

//@return 状态量

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

{



   return cv_verify_serialize_feature(p_feature,feature_str);

}



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

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

//@return 特征值

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

//@param 特征值

//@return 状态量

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

{



   p_feature = cv_verify_deserialize_feature(feature_str);

   if(p_feature != NULL){

   if(p_feature != NULL) {

      return 0;

   }else

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

}