此基本类用于检测 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. 检测结果
函数快速参考
下表列出了 vitis::ai::UltraFast
类中定义的所有函数:
类型 | 成员 | 实参 |
---|---|---|
std::unique_ptr<
UltraFast
> |
create |
|
vitis::ai::UltraFastResult
|
run |
|
std::vector<
vitis::ai::UltraFastResult
> |
run |
|
int | getInputWidth |
|
int | getInputHeight |
|
size_t | get_input_batch |
|