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

Vitis AI Library 用户指南 (UG1354)

Document ID
UG1354
Release Date
2023-06-29
Version
3.5 简体中文
此基本类用于检测 CULane 数据集的车道线。

输入为图像 (cv:Mat)。

输出是检测结果的结构体,名为 UltraFastResult

代码样本:

Mat img = cv::imread("sample_ultrafast.jpg");
auto ultrafast = vitis::ai::UltraFast::create("ultrafast_pt",true);
auto results = ultrafast->run(img);
for(const auto &lanes : results.lanes){
  std::cout <<"lane:\n";
  for(auto &v: lanes) {
     std::cout << "    ( " << v.first << ", " << v.second << " )\n";
  }
}

显示模型结果:

图 1. 检测结果
detection result

函数快速参考

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

表 1. 函数快速参考
类型 成员 实参
std::unique_ptr< UltraFast > create
  • const std::string & model_name
  • bool need_preprocess
vitis::ai::UltraFastResult run
  • const cv::Mat & img
std::vector< vitis::ai::UltraFastResult > run
  • const std::vector< cv::Mat > & imgs
int getInputWidth
  • void
int getInputHeight
  • void
size_t get_input_batch
  • void