mirror of
https://github.com/bububa/openvision.git
synced 2025-10-05 05:36:50 +08:00
32 lines
788 B
C++
32 lines
788 B
C++
#ifndef _POSE_DETECTER_H_
|
|
#define _POSE_DETECTER_H_
|
|
|
|
#include "../common/common.hpp"
|
|
#include <vector>
|
|
namespace ovpose {
|
|
|
|
class Detecter: public ov::Estimator {
|
|
public:
|
|
virtual int ExtractROIs(const unsigned char* rgbadata,
|
|
int img_width, int img_height,
|
|
std::vector<ov::ObjectInfo>* rois) = 0;
|
|
virtual int ExtractKeypoints(const unsigned char* rgbdata,
|
|
int img_width, int img_height,
|
|
const ov::Rect& rect, std::vector<ov::Keypoint>* keypoints) = 0;
|
|
};
|
|
|
|
class DetecterFactory {
|
|
public:
|
|
virtual Detecter* CreateDetecter() = 0;
|
|
virtual ~DetecterFactory() {};
|
|
};
|
|
|
|
class UltralightFactory: public DetecterFactory {
|
|
public:
|
|
UltralightFactory() {}
|
|
~UltralightFactory() {}
|
|
Detecter* CreateDetecter();
|
|
};
|
|
}
|
|
#endif // !_POSE_DETECTER_H
|