blob: 204af0b528d45f98e1504332021345ee3944a63a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
//
// Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once
#include "DetectedObject.hpp"
#include <numeric>
#include <vector>
namespace od
{
/**
* @brief Returns a vector of indices corresponding to input detections kept after NMS.
*
* Perform non max suppression on input detections. Any detections with iou greater than
* given threshold are suppressed. Different detection labels are considered independently.
*
* @param[in] Vector of decoded detections.
* @param[in] Detects with IOU larger than this threshold are suppressed.
* @return Vector of indices corresponding to input detections kept after NMS.
*
*/
std::vector<int> NonMaxSuppression(DetectedObjects& inputDetections, float iouThresh);
}// namespace od
|