//
|
// Created by pans on 4/27/18.
|
//
|
#include <iostream>
|
#include <string>
|
#include <basic/util/app/AppPreference.hpp>
|
#include <atomic>
|
|
#include "CasiaFaceWrapperN.h"
|
|
using namespace std;
|
|
std::atomic_int CasiaFaceWrapperN::instanceCount(0);
|
|
CasiaFaceWrapperN::CasiaFaceWrapperN() {
|
threadMax = appPref.getLongData("thread.max");
|
// long gpuIndex = appPref.getLongData("gpu.index");
|
// for(int i = 0;i<threadMax;i++){
|
// resourcesManager.pushResource(i);
|
// }
|
// if(gpuIndex < 0){
|
// THFI_Param* param = new THFI_Param[threadMax];
|
//// THFI_Create(threadMax, param);
|
// CHKERR(THFI_Create(threadMax, param), threadMax, "THFI_Create return");//old version ,default GPU device ID is 0
|
// CHKERR(EF_Init(threadMax), threadMax, "EF_Create return");//old version ,default GPU device ID is 0
|
// delete[] param;
|
// }else{
|
// THFI_Param_Ex* param = new THFI_Param_Ex[threadMax];
|
// THFI_Param detParam;
|
// EF_Param pParam[threadMax];
|
// detParam.nMinFaceSize = 20;
|
// detParam.nRollAngle = 60;
|
// for(int i = 0;i<threadMax;i++){
|
// param[i].tp = detParam;
|
// param[i].nDeviceID = gpuIndex;
|
// pParam[i].nDeviceID = gpuIndex;//GPU device ID,eg:0,1,2....
|
// }
|
// THFI_Create_Ex(threadMax, param);
|
// CHKERR(EF_Init_Ex(threadMax,pParam), threadMax, "EF_Create return");//old version ,default GPU device ID is 0
|
//// EF_Init_Ex(gpuIndexs.size(),pParam);
|
// delete[] param;
|
// }
|
// THFP_Create(threadMax);
|
|
channel = instanceCount;
|
THFI_Param detParam;
|
detParam.nMinFaceSize = 50;
|
detParam.nRollAngle = 60;
|
if(instanceCount++ == 0){
|
// if(gpuIndexs.empty()){
|
CHKERR(THFI_Create(threadMax, &detParam), threadMax, "THFI_Create return");//old version ,default GPU device ID is 0
|
CHKERR(EF_Init(threadMax), threadMax, "EF_Create return");//old version ,default GPU device ID is 0
|
// }
|
// else{
|
// THFI_Param_Ex pParamx[gpuIndexs.size()];
|
// EF_Param pParam[gpuIndexs.size()];
|
// for(int i = 0; i<gpuIndexs.size(); i++){
|
// pParamx[i].tp=detParam;
|
// pParamx[i].nDeviceID=gpuIndexs[i];//GPU device ID,eg:0,1,2....
|
// pParam[i].nDeviceID=gpuIndexs[i];//GPU device ID,eg:0,1,2....
|
// }
|
// int ret=THFI_Create_Ex(gpuIndexs.size(), pParamx);
|
// if(ret !=gpuIndexs.size()){
|
// printf("THFI_Create failed!(ret=%d)\n",ret);
|
// }
|
// ret=EF_Init_Ex(gpuIndexs.size(),pParam);
|
// if(ret !=gpuIndexs.size()){
|
// printf("EF_Init_Ex failed!(ret=%d)\n",ret);
|
// }
|
// }
|
}else if(instanceCount> threadMax){
|
channel = -1;
|
ERR("too many channels instanced, channel "<<channel<<" instance faild.");
|
}
|
|
}
|
|
CasiaFaceWrapperN::~CasiaFaceWrapperN() {
|
//#todo
|
THFI_Release();
|
THFP_Release();
|
}
|
|
float CasiaFaceWrapperN::compareFeature(std::vector<unsigned char> &feature1, std::vector<unsigned char> &feature2) {
|
if(feature1.size()!= EF_Size()){
|
ERR("feature1 size is not incorrect");
|
return 0;
|
}else if(feature2.size()!= EF_Size()){
|
ERR("feature2 size is not incorrect");
|
return 0;
|
}
|
return EF_Compare(feature1.data(), feature2.data());
|
}
|
#define COPY(NAME) memcpy(&face.NAME, &pos.NAME, sizeof (pos.NAME));s
|
Features CasiaFaceWrapperN::extractFace(FaceImageN img) {
|
Features results;
|
if(channel ==-1){
|
ERR("invalid face channel, face detect faild");
|
return results;
|
}
|
vector<unsigned char> feature;
|
THFI_FacePos facesPos[30];
|
// int channel;
|
// for(int loop = 0;loop < threadMax;loop++){
|
// channel = resourcesManager.getAvilableChannel(to_string(loop));
|
// if(channel > 0){
|
// cout << channel << endl;
|
// break;
|
// }
|
// }
|
int faceNum = THFI_DetectFace(channel, img.data,24, img.width, img.height, facesPos ,30);
|
if(faceNum>0){
|
if(feature.size()!= EF_Size()*faceNum){
|
feature.resize(EF_Size()*faceNum);
|
}
|
CHKERR(EF_Extract_M(channel, img.data, img.width, img.height, 3, facesPos, feature.data(), faceNum), 1, " return");
|
for(int i = 0; i< faceNum; i++){
|
FaceFeaWithScore result;
|
THFI_FacePos& face = facesPos[i];
|
result.left = face.rcFace.left;
|
result.top = face.rcFace.top;
|
result.width = face.rcFace.right - face.rcFace.left;
|
result.height = face.rcFace.bottom - face.rcFace.top;
|
result.score = face.fAngle.confidence;
|
// if(face.fAngle.confidence >= 0.6){
|
if((face.fAngle.pitch >-30 || face.fAngle.pitch <30 ) &&
|
(face.fAngle.roll >-30 || face.fAngle.roll <30 ) &&
|
(face.fAngle.yaw >-30 || face.fAngle.yaw <30 )){
|
result.feature.resize(EF_Size());
|
memcpy(result.feature.data(),feature.data()+i*EF_Size(),EF_Size());
|
results.push_back(result);
|
}
|
// }
|
}
|
} else{
|
DBG("faceNum is < 0");
|
}
|
return results;
|
}
|
|
|
//#todo
|
//Features CasiaFaceWrapperN::extractFace(FaceImageN &img, THFI_FacePos& facePos,int faceNum) {
|
// return Features();
|
//}
|
