此基本类用于检测来自输入图像 (cv::Mat) 的姿态。
输入图像 (cv::Mat)。
注释: 支持检测单一姿态。
输出为
PoseDetectResult
结构体,包含 14 个点。
代码样本:
auto det = vitis::ai::PoseDetect::create
("sp_net");
auto image = cv::imread("sample_posedetect.jpg");
auto results = det->run(image);
for(auto result: results.pose14pt) {
std::cout << result << std::endl;
}
显示 posedetect 模型结果:
图 1. 姿态检测图像
函数快速参考
下表列出了 vitis::ai::PoseDetect
类中定义的所有函数:
类型 | 成员 | 实参 |
---|---|---|
std::unique_ptr<
PoseDetect
> |
create |
|
int | getInputWidth |
|
int | getInputHeight |
|
size_t | get_input_batch |
|
PoseDetectResult
|
run |
|
std::vector<
PoseDetectResult
> |
run |
|