vitis::ai::PoseDetect - 3.5 简体中文

Vitis AI Library 用户指南 (UG1354)

Document ID
UG1354
Release Date
2023-06-29
Version
3.5 简体中文
此基本类用于检测来自输入图像 (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. 姿态检测图像
pose detect image

函数快速参考

下表列出了 vitis::ai::PoseDetect 类中定义的所有函数:

表 1. 函数快速参考
类型 成员 实参
std::unique_ptr< PoseDetect > create
  • const std::string & model_name
  • bool need_preprocess
int getInputWidth
  • void
int getInputHeight
  • void
size_t get_input_batch
  • void
PoseDetectResult run
  • const cv::Mat & image
std::vector< PoseDetectResult > run
  • const std::vector< cv::Mat > & images