#ifndef _PL_SENSETIMEFACETRACK_H_
|
#define _PL_SENSETIMEFACETRACK_H_
|
|
#include "PipeLine.h"
|
#include "GraphicHelper.h"
|
#include <vector>
|
#include <cmath>
|
|
struct SensetimeFaceFeature
|
{
|
PLGH_Rect rect;
|
int id;
|
|
float score;
|
|
/* Camera vision vector point to face
|
* * * *
|
* * * *
|
* * * *
|
*/
|
float yaw;
|
|
/* Camera vision vector point to face
|
* * * *
|
* * * *
|
* * * *
|
*/
|
float pitch;
|
|
/* Camera vision vector point to face
|
* * * *
|
* * * *
|
* * * *
|
*/
|
float roll;
|
|
float eyeDistance;
|
PLGH_Path featurePoints;
|
|
bool outOfFrame;
|
|
SensetimeFaceFeature() :
|
rect(), id(0), score(0.0), yaw(0.0), pitch(0.0), roll(0.0), eyeDistance(0.0), featurePoints(),
|
outOfFrame(false)
|
{}
|
|
bool test_face_in_cone(float _yaw, float _pitch, float _roll) const
|
{
|
return (std::abs(yaw) < _yaw && std::abs(pitch) < _pitch && std::abs(roll) < _roll);
|
}
|
};
|
|
typedef std::vector<SensetimeFaceFeature> st_ff_vect_t;
|
|
struct SensetimeFaceTrackConfig
|
{
|
int point_size; // 21 / 106
|
int point_size_config; // CV_DETECT_ENABLE_ALIGN_21 / CV_DETECT_ENABLE_ALIGN_106
|
int detect_face_cnt_limit; // -1
|
bool draw_face_rect;
|
bool draw_face_feature_point;
|
bool generate_face_feature;
|
bool generate_face_point;
|
int explode_feature_rect_x;
|
int explode_feature_rect_y;
|
bool clamp_feature_rect; // clamp fr width and height
|
int doTrackPerFrame;
|
|
std::string license_file_path;
|
std::string license_str;
|
|
float visionConeAngle;
|
|
bool evenWidthHeight;
|
|
float score_min;
|
|
SensetimeFaceTrackConfig() :
|
point_size(21), point_size_config(-1), detect_face_cnt_limit(-1),
|
draw_face_rect(true), draw_face_feature_point(true), generate_face_feature(false), generate_face_point(false),
|
explode_feature_rect_x(0), explode_feature_rect_y(0),
|
clamp_feature_rect(false), doTrackPerFrame(1),
|
license_file_path(), license_str(),
|
visionConeAngle(90.1), evenWidthHeight(true),
|
score_min(0.0f)
|
{ }
|
};
|
|
class PL_SensetimeFaceTrack : public PipeLineElem
|
{
|
public:
|
PL_SensetimeFaceTrack();
|
virtual ~PL_SensetimeFaceTrack();
|
|
virtual bool init(void* args);
|
virtual void finit();
|
|
virtual bool pay(const PipeMaterial& pm);
|
virtual bool gain(PipeMaterial& pm);
|
|
private:
|
static bool pay_breaker_MBFT_YUV(const PipeMaterial* pm, void* args);
|
|
private:
|
void* internal;
|
};
|
|
PipeLineElem* create_PL_SensetimeFaceTrack();
|
|
#endif
|
