Return the indices of the top N values of vector v. */
| 85 | |
| 86 | /* Return the indices of the top N values of vector v. */ |
| 87 | static std::vector<int> Argmax(const std::vector<float>& v, int N) |
| 88 | { |
| 89 | std::vector<std::pair<float, int>> pairs; |
| 90 | for(size_t i = 0; i < v.size(); ++i) |
| 91 | pairs.push_back(std::make_pair(v[i], i)); |
| 92 | std::partial_sort(pairs.begin(), pairs.begin() + N, pairs.end(), PairCompare); |
| 93 | |
| 94 | std::vector<int> result; |
| 95 | for(int i = 0; i < N; ++i) |
| 96 | result.push_back(pairs[i].second); |
| 97 | return result; |
| 98 | } |
| 99 | |
| 100 | /* Return the top N predictions. */ |
| 101 | std::vector<Prediction> Classifier::Classify(const cv::Mat& img, int N) |