mirror of
https://gitee.com/bianbu-linux/ai-support
synced 2025-04-24 22:27:13 -04:00
24 lines
672 B
C++
24 lines
672 B
C++
#ifndef SUPPORT_INCLUDE_TASK_VISION_POSE_ESTIMATION_TASK_H_
|
|
#define SUPPORT_INCLUDE_TASK_VISION_POSE_ESTIMATION_TASK_H_
|
|
|
|
#include <memory> // for: shared_ptr
|
|
#include <string>
|
|
|
|
#include "opencv2/opencv.hpp"
|
|
#include "task/vision/object_detection_types.h"
|
|
#include "task/vision/pose_estimation_types.h"
|
|
|
|
class poseEstimationTask {
|
|
public:
|
|
explicit poseEstimationTask(const std::string &filePath);
|
|
~poseEstimationTask() = default;
|
|
int getInitFlag();
|
|
PoseEstimationResult Estimate(const cv::Mat &raw_img, const Boxi &box);
|
|
|
|
private:
|
|
class impl;
|
|
std::shared_ptr<impl> pimpl_;
|
|
int init_flag_;
|
|
};
|
|
|
|
#endif // SUPPORT_INCLUDE_TASK_VISION_POSE_ESTIMATION_TASK_H_
|