From 0cdd8b3321c7fe62c41a1f18c080739aa5b009fa Mon Sep 17 00:00:00 2001
From: xuxiuxi <xuxiuxi@454eff88-639b-444f-9e54-f578c98de674>
Date: 星期三, 03 五月 2017 12:10:06 +0800
Subject: [PATCH]
---
FaceServer/STFaceCache.cpp | 286 ++++++++++++++++++++++++++++++++++++++++++++++++++++++---
1 files changed, 270 insertions(+), 16 deletions(-)
diff --git a/FaceServer/STFaceCache.cpp b/FaceServer/STFaceCache.cpp
index f57dc58..c603dca 100644
--- a/FaceServer/STFaceCache.cpp
+++ b/FaceServer/STFaceCache.cpp
@@ -7,18 +7,79 @@
#include <dirent.h>
#include "sample_face_search.h"
+#include <cv_face.h>
-struct STFaceContext
+#define FDP_FDR_INVALID FDP_FaceDetectResult(0, 0)
+#define ENABLE_AUTO_CREATE_STFACEDB
+#define ENABLE_SEARCH_IN_NEGATIVE_DBID
+#define ENABLE_SEARCH_IN_NEGATIVE_DBID
+#define RESULT_CONFIDENCE 0.70
+
+struct STFaceCacheContext
{
- int dbid;
- std::string dbfn;
- stface_handles handles;
+ cv_handle_t handle_verify;
+ cv_handle_t handle_detect;
+
+ STFaceCacheContext() : handle_verify(nullptr), handle_detect(nullptr)
+ {
+ }
};
-typedef std::map<int, STFaceContext> stface_ctx_map_t; // <dbid, ctx>
+struct STFaceDBContext
+{
+ int dbid;
+ std::string dbfname;
+ std::string dbfpath;
+ stface_handles handles;
+ bool dbLoadOK;
+
+ STFaceDBContext() : dbid(0), dbfname(), dbfpath(), handles(), dbLoadOK(false)
+ {}
+
+ bool init_db(const STFaceCacheContext& ctx)
+ {
+ handles.handle_verify = ctx.handle_verify;
+ handles.handle_detect = ctx.handle_detect;
+ if (handles.handle_db == nullptr)
+ {
+ dbLoadOK = stface_db_load(handles, db_full_path().c_str());
+ dbLoadOK &= (handles.handle_db != nullptr);
+ }
+ else
+ {
+ dbLoadOK = true;
+ }
+ return dbLoadOK;
+ }
+
+ void close_db()
+ {
+ stface_db_save(handles, db_full_path().c_str());
+
+ handles.handle_verify = nullptr;
+ handles.handle_detect = nullptr;
+ cv_verify_destroy_db(handles.handle_db);
+ handles.handle_db = nullptr;
+ dbLoadOK = false;
+ }
+
+ std::string db_full_path() const
+ {
+ if (dbfpath.empty() || dbfname.empty())
+ return "";
+
+ std::string fn(dbfpath);
+ fn.append("/");
+ fn.append(dbfname);
+ return fn;
+ }
+};
+
+typedef std::map<int, STFaceDBContext> stface_ctx_map_t; // <dbid, ctx>
STFaceCache::STFaceCache(const std::string& _stfacedbPath)
- : stfacedbPath(_stfacedbPath), stfaceModels(STFACESDK_BASE "/models"), _cacheContext(new stface_ctx_map_t)
+ : stfacedbPath(_stfacedbPath), stfaceModels(STFACESDK_BASE "/models"),
+ _dbContext(new stface_ctx_map_t), _cacheContext(new STFaceCacheContext)
{
LOG_INFO << "st face db: " << stfacedbPath << LOG_ENDL;
LOG_INFO << "st face sdk models: " << stfaceModels << LOG_ENDL;
@@ -26,13 +87,58 @@
STFaceCache::~STFaceCache()
{
- delete (stface_ctx_map_t*)_cacheContext;
+ delete (stface_ctx_map_t*)_dbContext;
+ _dbContext = nullptr;
+
+ delete (STFaceCacheContext*)_cacheContext;
_cacheContext = nullptr;
+}
+
+bool STFaceCache::init()
+{
+ STFaceCacheContext& cacheContext(*(STFaceCacheContext*)_cacheContext);
+
+ if (stface_init_license("./license.lic") != 0)
+ {
+ LOG_ERROR << "stface_init_license failed" << LOG_ENDL;
+ return false;
+ }
+
+ cv_face_algorithm_info();
+
+ cv_result_t cv_result = cv_face_create_detector(&cacheContext.handle_detect, nullptr, CV_DETECT_ENABLE_ALIGN_106);//CV_DETECT_ENABLE_ALIGN_21
+ if (cv_result != CV_OK)
+ {
+ LOGP(ERROR, "create detect handle failed, error code %d", cv_result);
+ return false;
+ }
+
+ std::string modelsFile(stfaceModels);
+ modelsFile.append("/verify.model");
+ cv_result = cv_verify_create_handle(&cacheContext.handle_verify, modelsFile.c_str());
+ if (cv_result != CV_OK)
+ {
+ LOGP(ERROR, "create verify handle failed, error code %d\n", cv_result);
+ return false;
+ }
+
+ return true;
+}
+
+void STFaceCache::finit()
+{
+ STFaceCacheContext& cacheContext(*(STFaceCacheContext*)_cacheContext);
+
+ cv_face_destroy_detector(cacheContext.handle_detect);
+ cv_verify_destroy_handle(cacheContext.handle_verify);
+ cacheContext.handle_detect = nullptr;
+ cacheContext.handle_verify = nullptr;
}
bool STFaceCache::load_dbs()
{
- stface_ctx_map_t& cacheContext(*(stface_ctx_map_t*)_cacheContext);
+ stface_ctx_map_t& dbContext(*(stface_ctx_map_t*)_dbContext);
+ STFaceCacheContext& cacheContext(*(STFaceCacheContext*)_cacheContext);
DIR* dp = opendir(stfacedbPath.c_str());
if (dp != NULL)
@@ -42,9 +148,18 @@
{
//puts(ep->d_name);
- STFaceContext ctx;
- ctx.dbfn = ep->d_name;
- ctx.dbid = strtol(ctx.dbfn.c_str(), nullptr, 10);
+ STFaceDBContext ctx;
+ ctx.dbfname = ep->d_name;
+ ctx.dbfpath = stfacedbPath;
+ ctx.dbid = strtol(ctx.dbfname.c_str(), nullptr, 10);
+
+ if (ctx.dbfname.find(".stfacedb") == std::string::npos)
+ continue;
+
+ if (ctx.dbid == 0)
+ continue;
+
+ dbContext.insert(std::make_pair(ctx.dbid, ctx));
}
closedir(dp);
}
@@ -53,16 +168,155 @@
LOG_ERROR << "Couldn't open the directory." << LOG_ENDL;
return false;
}
+
+ for (stface_ctx_map_t::iterator iterCtx = dbContext.begin(); iterCtx != dbContext.end(); ++iterCtx)
+ {
+ STFaceDBContext& ctx(iterCtx->second);
+ bool ret = ctx.init_db(cacheContext);
+ LOG_INFO << "init stface db dbid=" << ctx.dbid << ", ret=" << ret << LOG_ENDL;
+ }
+
+ return true;
+}
+
+void STFaceCache::close_dbs()
+{
+ stface_ctx_map_t& dbContext(*(stface_ctx_map_t*)_dbContext);
+
+ for (stface_ctx_map_t::iterator iterCtx = dbContext.begin(); iterCtx != dbContext.end(); ++iterCtx)
+ {
+ STFaceDBContext& ctx(iterCtx->second);
+ ctx.close_db();
+ }
+}
+
+FDP_FaceDetectResult STFaceCache::detect_neg(const STFaceImage& img)
+{
+ if (img.db_id <= 0)
+ {
+ return FDP_FDR_INVALID;
+ }
+
+#ifdef ENABLE_SEARCH_IN_NEGATIVE_DBID
+ STFaceImage imgNeg(img);
+ imgNeg.db_id = -1;
+ return detect(imgNeg);
+#else
+ return FDP_FDR_INVALID;
+#endif
}
FDP_FaceDetectResult STFaceCache::detect(const STFaceImage& img)
{
- stface_ctx_map_t& cacheContext(*(stface_ctx_map_t*)_cacheContext);
- return FDP_FaceDetectResult(0, 0);
+ stface_ctx_map_t& dbContext(*(stface_ctx_map_t*)_dbContext);
+
+ LOG_WARN << "detect in dbid=" << img.db_id << LOG_ENDL;
+
+ if (img.db_id == 0)
+ {
+ // search all db
+ //#todo
+ }
+ else
+ {
+ stface_ctx_map_t::iterator iterCtx = dbContext.find(img.db_id);
+ if (iterCtx == dbContext.end())
+ {
+ LOG_WARN << "no db find dbid=" << img.db_id << LOG_ENDL;
+ return detect_neg(img);
+ }
+
+ STFaceDBContext& ctx(iterCtx->second);
+ if (!ctx.dbLoadOK)
+ {
+ LOG_WARN << "dbLoadOK return false" << LOG_ENDL;
+ return detect_neg(img);
+ }
+
+ top_idx_score_vect_t result;
+ if (!stface_search_db(ctx.handles, img, result))
+ {
+ LOG_WARN << "stface_search_db return false" << LOG_ENDL;
+ return detect_neg(img);
+ }
+
+ if (result.empty())
+ {
+ LOG_INFO << "stface_search_db return empty" << LOG_ENDL;
+ return detect_neg(img);
+ }
+
+ LOGP(INFO, "stface_search_db return dbid=%d, idx=%d, score=%f", img.db_id, result[0].idx, result[0].score);
+
+ if (result[0].score < RESULT_CONFIDENCE)
+ {
+ LOGP(INFO, "low score not accepted");
+ return detect_neg(img);
+ }
+
+ return FDP_FaceDetectResult(img.db_id, result[0].idx);
+ }
+
+ return FDP_FDR_INVALID;
}
-FDP_FaceDetectResult STFaceCache::save(const STFaceImage& img)
+FDP_FaceDetectResult STFaceCache::add(const STFaceImage& img)
{
- stface_ctx_map_t& cacheContext(*(stface_ctx_map_t*)_cacheContext);
- return FDP_FaceDetectResult(0, 0);
+ stface_ctx_map_t& dbContext(*(stface_ctx_map_t*)_dbContext);
+ STFaceCacheContext& cacheContext(*(STFaceCacheContext*)_cacheContext);
+
+ LOG_WARN << "add in dbid=" << img.db_id << LOG_ENDL;
+
+ if (img.db_id == 0)
+ {
+ LOG_WARN << "db_id=0 not ok" << LOG_ENDL;
+ return FDP_FDR_INVALID;
+ }
+
+ stface_ctx_map_t::iterator iterCtx = dbContext.find(img.db_id);
+
+#ifdef ENABLE_AUTO_CREATE_STFACEDB
+ if (iterCtx == dbContext.end())
+ {
+ // create db
+ STFaceDBContext ctx;
+ char dbfname[50];
+ sprintf(dbfname, "%d.stfacedb", img.db_id);
+ ctx.dbfname = dbfname;
+ ctx.dbfpath = stfacedbPath;
+ ctx.dbid = img.db_id;
+ ctx.init_db(cacheContext); // copy handlers
+
+ if (!stface_db_create(ctx.handles, ctx.db_full_path().c_str()))
+ {
+ LOG_WARN << "stface_db_create return false" << LOG_ENDL;
+ return FDP_FDR_INVALID;
+ }
+
+ ctx.dbLoadOK = true;
+
+ dbContext.insert(std::make_pair(ctx.dbid, ctx));
+ iterCtx = dbContext.find(img.db_id);
+ }
+#endif
+
+ if (iterCtx == dbContext.end())
+ {
+ LOG_ERROR << "no stfacedb found" << LOG_ENDL;
+ return FDP_FDR_INVALID;
+ }
+
+ STFaceDBContext& ctx(iterCtx->second);
+ if (!ctx.dbLoadOK)
+ {
+ LOG_WARN << "dbLoadOK return false" << LOG_ENDL;
+ return FDP_FDR_INVALID;
+ }
+
+ int idx = stface_db_add(ctx.handles, img);
+ LOG_INFO << "stface_db_add dbid=" << img.db_id << ", idx=" << idx << LOG_ENDL;
+ if (idx <= 1)
+ return FDP_FDR_INVALID;
+ else
+ return FDP_FaceDetectResult(img.db_id, idx);
}
--
Gitblit v1.8.0