提交 3e302147 编写于 作者: D dongshuilong

add print class_id for cls cpp infer

上级 71ab42bc
...@@ -103,7 +103,8 @@ int main(int argc, char **argv) { ...@@ -103,7 +103,8 @@ int main(int argc, char **argv) {
std::cout << "Current total inferen time cost: " << infer_time << " ms." std::cout << "Current total inferen time cost: " << infer_time << " ms."
<< std::endl; << std::endl;
for (int i = 0; i < max_len; ++i) { 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) if (label_output_equal_flag)
printf("label: %s\n", config.id_map[result_index[i]].c_str()); printf("label: %s\n", config.id_map[result_index[i]].c_str());
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册