Commit ac3e44ec authored by Ted Themistokleous's avatar Ted Themistokleous
Browse files

Add compare function to sort() on in filter_boxes_by_score()

This wasn't sorting correctly without this and failing other tests.
parent 14618b67
...@@ -253,7 +253,9 @@ struct nonmaxsuppression ...@@ -253,7 +253,9 @@ struct nonmaxsuppression
return std::make_pair(sc, box_idx - 1); return std::make_pair(sc, box_idx - 1);
}); });
} }
std::sort(std::execution::par, boxes_heap.begin(), boxes_heap.end()); std::sort(std::execution::par, boxes_heap.begin(), boxes_heap.end(), [&](auto a, auto b) {
return a.first > b.first;
});
return boxes_heap; return boxes_heap;
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment