feat(pose): add utralight posenet

This commit is contained in:
Syd Xu
2021-10-26 20:38:31 +08:00
parent 17ebe05f9c
commit 5d0fc1244f
34 changed files with 542 additions and 141 deletions

View File

@@ -0,0 +1,31 @@
#ifndef _POSE_DETECTER_H_
#define _POSE_DETECTER_H_
#include "../common/common.hpp"
#include <vector>
namespace ov {
class Detecter {
public:
virtual ~Detecter(){};
virtual int LoadModel(const char* root_path) = 0;
virtual int ExtractROIs(const unsigned char* rgbadata,
int img_width, int img_height,
std::vector<ROI>* rois) = 0;
virtual int ExtractKeypoints(const ROI& roi, std::vector<Keypoint>* keypoints) = 0;
};
class DetecterFactory {
public:
virtual Detecter* CreateDetecter() = 0;
virtual ~DetecterFactory() {};
};
class UtralightFactory: public DetecterFactory {
public:
UtralightFactory() {}
~UtralightFactory() {}
Detecter* CreateDetecter();
};
}
#endif // !_POSE_DETECTER_H