fix: pfld results uses forward list

Seeed-Studio · Dec 7, 2024 · 3eddbc4 · 3eddbc4
1 parent 6d0cf9d
commit 3eddbc4
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/SSCMA_Micro_Core.cpp b/src/SSCMA_Micro_Core.cpp
@@ -345,9 +345,8 @@ SSCMAMicroCore::Expected SSCMAMicroCore::invoke(const Frame& frame, const Invoke
             }
             auto results = algorithm->getResults();
             if (_config.invoke_config && _config.invoke_config->top_k > 0) {
-                std::sort(results.begin(), results.end(), [](const ma_point_t& a, const ma_point_t& b) { return a.score > b.score; });
-                results.resize(std::min(results.size(), static_cast<size_t>(_config.invoke_config->top_k)));
-                results.shrink_to_fit();
+                results.sort([](const ma_point_t& a, const ma_point_t& b) { return a.score < b.score; });
+                results.resize(_config.invoke_config->top_k);
             }
             std::vector<SSCMAMicroCore::Point> points;
             for (const auto& result : results) {