#ifndef _PL_SENSETIMEFACEDETECT_H_
|
#define _PL_SENSETIMEFACEDETECT_H_
|
|
#include "PipeLine.h"
|
#include "MaterialBuffer.h"
|
//#include "PL_SensetimeFaceTrack.h" // for st_ff_vect_t
|
|
struct SensetimeFaceDetectDbFrame : public MB_Frame
|
{
|
int school_id;
|
void* _faceDB;
|
|
SensetimeFaceDetectDbFrame() : MB_Frame(), school_id(-1), _faceDB(nullptr)
|
{ }
|
};
|
|
struct SensetimeFaceDetectResult
|
{
|
int school_id;
|
int st_id;
|
//st_ff_vect_t features;
|
|
SensetimeFaceDetectResult() : school_id(-1), st_id(-1)
|
{ }
|
};
|
|
struct PL_SensetimeFaceDetectConfig
|
{
|
// #todo currently only support default config
|
bool payWithDbFrame;
|
bool resultStructOnly;
|
|
//std::string dbPath;
|
|
PL_SensetimeFaceDetectConfig() :
|
payWithDbFrame(true), resultStructOnly(true)
|
{ }
|
};
|
|
class PL_SensetimeFaceDetect : public PipeLineElem
|
{
|
public:
|
PL_SensetimeFaceDetect();
|
virtual ~PL_SensetimeFaceDetect();
|
|
virtual bool init(void* args);
|
virtual void finit();
|
|
virtual bool pay(const PipeMaterial& pm);
|
virtual bool gain(PipeMaterial& pm);
|
|
private:
|
void* internal;
|
};
|
|
PipeLineElem* create_PL_SensetimeFaceDetect();
|
|
#endif
|