57 lines
1.7 KiB
C++
57 lines
1.7 KiB
C++
|
|
#pragma once
|
|
#include <vector>
|
|
#include <memory>
|
|
#include "shapes/vp_rect.h"
|
|
|
|
|
|
namespace vp_objects {
|
|
// target in frame detected by face detectors such as yunet.
|
|
// note: we can define new target type like vp_frame_xxx_target... if need (see vp_frame_pose_target also)
|
|
class vp_frame_face_target
|
|
{
|
|
private:
|
|
/* data */
|
|
public:
|
|
vp_frame_face_target(int x,
|
|
int y,
|
|
int width,
|
|
int height,
|
|
float score,
|
|
std::vector<std::pair<int, int>> key_points = std::vector<std::pair<int, int>>(),
|
|
std::vector<float> embeddings = std::vector<float>());
|
|
~vp_frame_face_target();
|
|
|
|
// x of top left
|
|
int x;
|
|
// y of top left
|
|
int y;
|
|
// width of rect
|
|
int width;
|
|
// height of rect
|
|
int height;
|
|
|
|
// confidence
|
|
float score;
|
|
|
|
// feature vector created by infer nodes such as vp_sface_feature_encoder_node.
|
|
// embeddings can be used for face recognize or other reid works.
|
|
std::vector<float> embeddings;
|
|
|
|
// key points (5 points or more)
|
|
std::vector<std::pair<int, int>> key_points;
|
|
|
|
// track id filled by vp_track_node (child class) if it exists.
|
|
int track_id = -1;
|
|
// cache of track rects in the previous frames, filled by track node if it exists.
|
|
// we can draw / analyse depend on these track rects later.
|
|
std::vector<vp_objects::vp_rect> tracks;
|
|
|
|
// clone myself
|
|
std::shared_ptr<vp_frame_face_target> clone();
|
|
|
|
// rect area of target
|
|
vp_rect get_rect() const;
|
|
};
|
|
|
|
} |