add print class_id for cls cpp infer

pull/1557/head
dongshuilong 2021-12-14 06:39:47 +00:00
parent 71ab42bc82
commit 3e30214702
1 changed files with 2 additions and 1 deletions

View File

@ -103,7 +103,8 @@ int main(int argc, char **argv) {
std::cout << "Current total inferen time cost: " << infer_time << " ms."
<< std::endl;
for (int i = 0; i < max_len; ++i) {
printf("\tTop%d: score: %.4f, ", i + 1, out_data[result_index[i]]);
printf("\tTop%d: class_id: %d, score: %.4f, ", i + 1, result_index[i],
out_data[result_index[i]]);
if (label_output_equal_flag)
printf("label: %s\n", config.id_map[result_index[i]].c_str());
}