You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
788 B
28 lines
788 B
//
|
|
// 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
|