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

Vitis AI Library 用户指南 (UG1354)

Document ID
UG1354
Release Date
2023-06-29
Version
3.5 简体中文
此基本类用于检测输入图像 (cv::Mat) 中的对象。

输入为图像 (cv::Mat)。

输出是输入图像中对象的索引和得分。

代码样本:

auto image = cv::imread("sample_vehicleclassification.jpg");
auto network = vitis::ai::VehicleClassification::create(
               "vehicle_type_resnet18_pt");
auto result = network->run(image);
for (const auto &r : result.scores) {
   auto score = r.score;
   auto index = result.lookup(r.index);
}

函数快速参考

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

表 1. 函数快速参考
类型 成员 实参
std::unique_ptr< VehicleClassification > create
  • const std::string & model_name
  • bool need_preprocess
vitis::ai::VehicleClassificationResult run
  • const cv::Mat & image
std::vector< vitis::ai::VehicleClassificationResult > run
  • const std::vector< cv::Mat > & images