"git@developer.sourcefind.cn:gaoqiong/migraphx.git" did not exist on "7dd07b88a67b64d033ff16ccd0ba0946ecc9ed09"
Commit d6a2cfb5 authored by charlie's avatar charlie
Browse files

Formatting

parent 41663f68
......@@ -129,11 +129,8 @@ struct nonmaxsuppression
// filter boxes below score_threshold
template <class T>
std::priority_queue<std::pair<float, int64_t>> filter_boxes_by_score(T scores,
std::size_t score_offset_ind,
std::size_t num_boxes,
float score_threshold
) const
std::priority_queue<std::pair<float, int64_t>> filter_boxes_by_score(
T scores, std::size_t score_offset_ind, std::size_t num_boxes, float score_threshold) const
{
std::priority_queue<std::pair<float, int64_t>> boxes_heap;
auto insert_to_boxes_heap =
......@@ -185,8 +182,8 @@ struct nonmaxsuppression
(batch_idx * num_classes + class_idx) * num_boxes;
// index to first value of this batch
std::size_t batch_boxes_ind = batch_idx * num_boxes * 4;
auto boxes_heap = filter_boxes_by_score(
scores, score_offset_ind, num_boxes, score_threshold);
auto boxes_heap =
filter_boxes_by_score(scores, score_offset_ind, num_boxes, score_threshold);
selected_boxes_inside_class.clear();
// Get the next box with top score, filter by iou_threshold
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