Skip to content

GitLab

Commit c71201f9 authored by charlie's avatar charlie
Browse files

Formatting

parent 571a3464
Show whitespace changes
Inline Side-by-side
Showing with 22 additions and 22 deletions
src/include/migraphx/op/nonmaxsuppression.hpp
View file @ c71201f9
...@@ -129,8 +129,7 @@ struct nonmaxsuppression ...@@ -129,8 +129,7 @@ struct nonmaxsuppression
// filter boxes below score_threshold // filter boxes below score_threshold
template <class T> template <class T>
void filter_boxes_by_score( void filter_boxes_by_score(T scores,
T scores,
std::size_t score_offset_ind, std::size_t score_offset_ind,
std::size_t num_boxes, std::size_t num_boxes,
const float score_threshold, const float score_threshold,
...@@ -185,7 +184,8 @@ struct nonmaxsuppression ...@@ -185,7 +184,8 @@ struct nonmaxsuppression
// index to first value of this batch // index to first value of this batch
std::size_t batch_boxes_ind = batch_idx * num_boxes * 4; std::size_t batch_boxes_ind = batch_idx * num_boxes * 4;
std::priority_queue<std::pair<float, int64_t>> boxes_heap; std::priority_queue<std::pair<float, int64_t>> boxes_heap;
filter_boxes_by_score(scores, score_offset_ind, num_boxes, score_threshold, boxes_heap); filter_boxes_by_score(
scores, score_offset_ind, num_boxes, score_threshold, boxes_heap);
selected_boxes_inside_class.clear(); selected_boxes_inside_class.clear();
// Get the next box with top score, filter by iou_threshold // Get the next box with top score, filter by iou_threshold
while(!boxes_heap.empty() && while(!boxes_heap.empty() &&
......
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