mirror of
https://github.com/Qortal/Brooklyn.git
synced 2025-02-21 06:35:53 +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
|