From 9ee90782bf14b91442cd524bc033bbf6d1778bb2 Mon Sep 17 00:00:00 2001 From: houxiao <houxiao@454eff88-639b-444f-9e54-f578c98de674> Date: 星期四, 29 六月 2017 10:32:05 +0800 Subject: [PATCH] aaaaa --- FaceServer/main_face_daemon.cpp | 36 ++++++++++++++++++++++++++++++++++++ 1 files changed, 36 insertions(+), 0 deletions(-) diff --git a/FaceServer/main_face_daemon.cpp b/FaceServer/main_face_daemon.cpp index 039ec87..bd12356 100644 --- a/FaceServer/main_face_daemon.cpp +++ b/FaceServer/main_face_daemon.cpp @@ -190,6 +190,39 @@ return send_SensetimeFaceDetectResultJson(client, result, 0); } +bool ev_proc_SensetimeFaceDetectSearch(EVClientStub& client) +{ + EVPHeader* evpHeader = (EVPHeader*)client.recvBuff; + FDP_Image* fdpImage = (FDP_Image*)(client.recvBuff + sizeof(EVPHeader)); + fdpImage->ntoh(); + + STFaceImage stfaceImg; + stfaceImg.db_id = fdpImage->db_id; + stfaceImg.mb_type = fdpImage->mb_type; + stfaceImg.width = fdpImage->width; + stfaceImg.height = fdpImage->height; + stfaceImg.size = evpHeader->size - sizeof(EVPHeader) - sizeof(FDP_Image); + stfaceImg.buff = fdpImage->buff; + + LOGP(DEBUG, "stfaceImg db_id=%d, mb_type=%d, width=%d, height=%d, size=%d", + (int)stfaceImg.db_id, (int)stfaceImg.mb_type, (int)stfaceImg.width, (int)stfaceImg.height, (int)stfaceImg.size); + + + char imgfn[100 * 1024]; + static int i = 0; + sprintf(imgfn, "IMG_%d_%d_w%d_h%d.rgb565", stfaceImg.db_id, ++i, stfaceImg.width, stfaceImg.height); + FILE * pFile = fopen(imgfn, "wb"); + fwrite(stfaceImg.buff, sizeof(char), stfaceImg.size, pFile); + fclose(pFile); + pFile = nullptr; + + fdr_vec_t result; + g_STFaceCache.search(stfaceImg, result); + + //int ret = (fdrResult.db_id == 0 ? -1 : 0); + return send_SensetimeFaceDetectResultJson(client, result, 0); +} + bool ev_dispatcher_proto_pb(EVClientStub& client) { LOG_DEBUG << "ev_dispatcher_proto_pb" << LOG_ENDL; @@ -237,6 +270,9 @@ case FaceDaemonCommand::FDC_SENSETIMEFACEDETECT_COMPARE: return ev_proc_SensetimeFaceDetectCompare(client); break; + case FaceDaemonCommand::FDC_SENSETIMEFACEDETECT_SEARCH: + return ev_proc_SensetimeFaceDetectSearch(client); + break; default: LOG_WARN << "Unknown command" << LOG_ENDL; ev_send_status_packet(client, EVPStatus::EVPS_COMMAND_ERROR); -- Gitblit v1.8.0