Function Argmax

examples/tengine_model/classification/classification_fp16.cpp:106–117

}

static inline std::vector<int> Argmax(const std::vector<__fp16>& v, int N)
{
    std::vector<std::pair<__fp16, int>> pairs;
    for(size_t i = 0; i < v.size(); ++i)
        pairs.push_back(std::make_pair(v[i], i));
    std::partial_sort(pairs.begin(), pairs.begin() + N, pairs.end(), PairCompare);

    std::vector<int> result;
    for(int i = 0; i < N; ++i)
        result.push_back(pairs[i].second);
    return result;
}

void get_input_data(const char* image_file, __fp16* input_data, int img_h, int img_w, const float* mean, float scale)
{