多任务网络类型,声明 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
类中定义的所有函数:
类型 | 成员 | 实参 |
---|---|---|
std::unique_ptr< MultiTask > | create |
|
int | getInputWidth |
|
int | getInputHeight |
|
size_t | get_input_batch |
|
MultiTaskResult
|
run_8UC1 |
|
std::vector<
MultiTaskResult
> |
run_8UC1 |
|
MultiTaskResult
|
run_8UC3 |
|
std::vector<
MultiTaskResult
> |
run_8UC3 |
|