xingzilong
2017-08-18 9e5babf9db52e64bdae60137be7696e56241fca6
RtspFace/PL_SensetimeFaceDetect.cpp
@@ -2,8 +2,8 @@
#include "MaterialBuffer.h"
#include "logger.h"
#include "SensetimeFaceAPIWrapper/src/FaceDBPool.h"
#include "SensetimeFaceAPIWrapper/src/faceAPI.h"
//#include "SensetimeFaceAPIWrapper/src/FaceDBPool.h"
//#include "SensetimeFaceAPIWrapper/src/faceAPI.h"
#include <opencv2/opencv.hpp>
#include <cv_face.h>
@@ -95,12 +95,12 @@
      return false;
   }
   FaceDB* _faceDB = (FaceDB*)frame->_faceDB;
   if (_faceDB == nullptr)
   {
      LOG_ERROR << "FaceDB is null" << std::endl;
      return false;
   }
   //FaceDB* _faceDB = (FaceDB*)frame->_faceDB;
   //if (_faceDB == nullptr)
   //{
   //   LOG_ERROR << "FaceDB is null" << std::endl;
   //   return false;
   //}
   
   const size_t expectedYUVSize = frame->width * frame->height * 1.5;
   if (frame->buffSize < expectedYUVSize)
@@ -110,8 +110,9 @@
   }
   
   cv::Mat yuvMat(cv::Size(frame->width,frame->height), CV_8UC3, frame->buffer);
   cv_feature_t* feature = _faceDB->extract_feature(yuvMat);
   in->lastResult.st_id = _faceDB->search_db(feature);
   //cv_feature_t* feature = _faceDB->extract_feature(yuvMat);
   //in->lastResult.st_id = _faceDB->search_db(feature);
   //in->lastResult.st_id = _faceDB->do_reasch(yuvMat);
   //#todo release feature