mirror of
https://github.com/Qortal/Brooklyn.git
synced 2025-02-13 10:45:54 +00:00
28 lines
788 B
C++
28 lines
788 B
C++
//
|
|
// 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
|