vitis::ai::MultiTask - 2.5 简体中文

Vitis AI Library 用户指南 (UG1354)

Document ID
UG1354
Release Date
2022-06-15
Version
2.5 简体中文
多任务网络类型,声明 multitask 网络。

类数量标签:0;名称:"background";标签:1;名称:"person";标签:2;名称:"car";标签:3;名称:"truck";标签:4;名称:"bus";标签:5;名称:"bike";标签:6;名称:"sign";标签:7;名称:"light"

此基本类对应来自图像 (cv::Mat) 的 ADAS MuiltTask。

输入图像 (cv::Mat)。

输出是 MultiTaskResult 结构体,包含分割结果、检测结果和车辆朝向;

代码样本:

auto det = vitis::ai::MultiTask::create("multi_task");
auto image = cv::imread("sample_multitask.jpg");
auto result = det->run_8UC3(image);
cv::imwrite("sample_multitask_result.jpg",result.segmentation);

显示模型结果:

图 1. 结果图像

函数快速参考

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

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