此基本类用于检测输入图像 (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
类中定义的所有函数:
类型 | 成员 | 实参 |
---|---|---|
std::unique_ptr<
VehicleClassification
> |
create |
|
vitis::ai::VehicleClassificationResult
|
run |
|
std::vector<
vitis::ai::VehicleClassificationResult
> |
run |
|