From d9125b1f5e3bfd1c657d75c4ef812f1202dc406d Mon Sep 17 00:00:00 2001
From: houxiao <houxiao@454eff88-639b-444f-9e54-f578c98de674>
Date: 星期一, 10 七月 2017 10:12:44 +0800
Subject: [PATCH] proto support st_track_id
---
FaceServer/STFaceCache.cpp | 61 ++++++++++++++++++++++--------
1 files changed, 45 insertions(+), 16 deletions(-)
diff --git a/FaceServer/STFaceCache.cpp b/FaceServer/STFaceCache.cpp
index 52e3c69..6d5026d 100644
--- a/FaceServer/STFaceCache.cpp
+++ b/FaceServer/STFaceCache.cpp
@@ -9,12 +9,6 @@
#include "sample_face_search.h"
#include <cv_face.h>
-#define FDP_FDR_INVALID FDP_FaceDetectResult(0, 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
{
cv_handle_t handle_verify;
@@ -72,7 +66,7 @@
handles.handle_db = nullptr;
dbLoadOK = false;
}
-
+
std::string db_full_path() const
{
if (dbfpath.empty() || dbfname.empty())
@@ -89,7 +83,8 @@
STFaceCache::STFaceCache(const std::string& _stfacedbPath)
: stfacedbPath(_stfacedbPath), stfaceModels(STFACESDK_BASE "/models"),
- _dbContext(new stface_ctx_map_t), _cacheContext(new STFaceCacheContext)
+ _dbContext(new stface_ctx_map_t), _cacheContext(new STFaceCacheContext),
+ tempdbFaceCount(0)
{
LOG_INFO << "st face db: " << stfacedbPath << LOG_ENDL;
LOG_INFO << "st face sdk models: " << stfaceModels << LOG_ENDL;
@@ -207,13 +202,22 @@
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;
+#ifdef ENABLE_DETECT_IN_NEGATIVE_DBID
+ STFaceImage imgNeg(img);
+ FDP_FaceDetectResult res(FDP_FDR_INVALID);
+
+ imgNeg.db_id = STFS_DBID_VISITOR_1;
+ res = detect(imgNeg); // #todo optimize extract feature once
+ if (res.db_id != STFS_DBID_INVALID)
+ return res;
+
+ imgNeg.db_id = STFS_DBID_TEMPDB_1;
+ res = detect(imgNeg);
+ if (res.db_id != STFS_DBID_INVALID)
+ return res;
#endif
+
+ return FDP_FDR_INVALID;
}
FDP_FaceDetectResult STFaceCache::detect(const STFaceImage& img)
@@ -323,6 +327,10 @@
return FDP_FDR_INVALID;
}
+#ifdef ENABLE_ADD_TO_TEMPDB_WHEN_NOT_DETECT
+ tempdbFaceCount++;
+#endif
+
int idx = stface_db_add(ctx.handles, img);
LOG_INFO << "stface_db_add dbid=" << img.db_id << ", idx=" << idx << LOG_ENDL;
if (idx <= 1)
@@ -347,7 +355,6 @@
stface_ctx_map_t& dbContext(*(stface_ctx_map_t*)_dbContext);
STFaceCacheContext& cacheContext(*(STFaceCacheContext*)_cacheContext);
-
for(stface_ctx_map_t::iterator iterCtx = dbContext.begin(); iterCtx != dbContext.end(); ++iterCtx)
{
STFaceDBContext& ctx(iterCtx->second);
@@ -416,4 +423,26 @@
topResult.push_back(FDP_FaceDetectResult(-1, 2, 4));
topResult.push_back(FDP_FaceDetectResult(1, 6, 5));
*/
-}
\ No newline at end of file
+}
+
+void STFaceCache::delete_db(int dbid)
+{
+ stface_ctx_map_t& dbContext(*(stface_ctx_map_t*)_dbContext);
+ STFaceCacheContext& cacheContext(*(STFaceCacheContext*)_cacheContext);
+
+ stface_ctx_map_t::iterator iterCtx = dbContext.find(dbid);
+ if (iterCtx == dbContext.end())
+ {
+ LOG_WARN << "no dbid=%d" << dbid << LOG_ENDL;
+ return;
+ }
+
+ STFaceDBContext& ctx(iterCtx->second);
+ ctx.close_db();
+
+ char mvDBtoBak[1000];
+ sprintf(mvDBtoBak, "cd %s; mv %s %d.bak;", stfacedbPath.c_str(), ctx.db_full_path().c_str(), dbid);
+ system(mvDBtoBak);
+
+ dbContext.erase(iterCtx);
+}
--
Gitblit v1.8.0