Function PrintTopLabels

examples/tensorflow_wrapper/label_image/label_image.cpp:139–167

}

static int PrintTopLabels(const vector<TF_Tensor*>& outputs, const char* labels_file)
{
    vector<string> labels;
    int read_labels_status = ReadLabelsFile(labels_file, &labels);
    if(read_labels_status < 0)
        return -1;

    int label_count = labels.size();
    int N = std::min<int>(label_count, 5);

    float* data = ( float* )TF_TensorData(outputs[0]);

    vector<pair<int, float>> scores;
    for(int i = 0; i < label_count; i++)
    {
        scores.push_back(pair<int, float>({i, data[i]}));
    }

    sort(scores.begin(), scores.end(),
         [](const pair<int, float>& left, const pair<int, float>& right) { return left.second > right.second; });

    for(int pos = 0; pos < N; pos++)
    {
        const int label_index = scores[pos].first;
        const float score = scores[pos].second;
        cout << std::fixed << std::setprecision(4) << score << " - \"" << labels[label_index] << "\"" << endl;
    }
    return 0;
}

/* To make tensor release happy...*/
static void dummy_deallocator(void* data, size_t len, void* arg) {}