vitis::ai::Classification - 1.2 English

Input is an image (cv::Mat).

Output is index and score of objects in the input image.

Sample code:

auto image = cv::imread("sample_classification.jpg");
auto network = vitis::ai::Classification::create(
               "resnet_50",
               true);
auto result = network->run(image);
for (const auto &r : result.scores) {
   auto score = r.score;
   auto index = network->lookup(r.index);
}

Quick Function Reference

The following table lists all the functions defined in the vitis::ai::Classification class: