From 22d3d8fccb06812fc20e0cfcf96cb9c241fd9e52 Mon Sep 17 00:00:00 2001
From: pans <pans@454eff88-639b-444f-9e54-f578c98de674>
Date: 星期二, 03 一月 2017 18:47:49 +0800
Subject: [PATCH]
---
RtspFace/demo/src/faceAPI.cpp | 173 ++++++++++++++++++++++++++++++++++++++++++
RtspFace/demo/src/faceAPI.h | 36 +++++++++
2 files changed, 206 insertions(+), 3 deletions(-)
diff --git a/RtspFace/demo/src/faceAPI.cpp b/RtspFace/demo/src/faceAPI.cpp
index c65f15a..979895f 100644
--- a/RtspFace/demo/src/faceAPI.cpp
+++ b/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 鍦ㄦ暟鎹簱涓悳绱汉鑴竔d
+//@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)
{
@@ -74,4 +87,158 @@
return 0;
} 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;
}
\ No newline at end of file
diff --git a/RtspFace/demo/src/faceAPI.h b/RtspFace/demo/src/faceAPI.h
index 7148f09..c41b0d8 100644
--- a/RtspFace/demo/src/faceAPI.h
+++ b/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
--
Gitblit v1.8.0