24.04
|
CPP Function to perform non maximum suppression on the bounding boxes and scores. More...
#include <CPPNonMaximumSuppression.h>
Public Member Functions | |
void | configure (const ITensor *bboxes, const ITensor *scores, ITensor *indices, unsigned int max_output_size, const float score_threshold, const float nms_threshold) |
Configure the function to perform non maximal suppression. More... | |
Public Member Functions inherited from ICPPSimpleFunction | |
ICPPSimpleFunction () | |
Constructor. More... | |
void | run () override final |
Run the kernels contained in the function. More... | |
Public Member Functions inherited from IFunction | |
virtual | ~IFunction ()=default |
Destructor. More... | |
virtual void | prepare () |
Prepare the function for executing. More... | |
Static Public Member Functions | |
static Status | validate (const ITensorInfo *bboxes, const ITensorInfo *scores, const ITensorInfo *indices, unsigned int max_output_size, const float score_threshold, const float nms_threshold) |
Static function to check if given arguments will lead to a valid configuration of CPPNonMaximumSuppression. More... | |
CPP Function to perform non maximum suppression on the bounding boxes and scores.
Definition at line 37 of file CPPNonMaximumSuppression.h.
void configure | ( | const ITensor * | bboxes, |
const ITensor * | scores, | ||
ITensor * | indices, | ||
unsigned int | max_output_size, | ||
const float | score_threshold, | ||
const float | nms_threshold | ||
) |
Configure the function to perform non maximal suppression.
[in] | bboxes | The input bounding boxes. Data types supported: F32. |
[in] | scores | The corresponding input confidence. Same as bboxes . |
[out] | indices | The kept indices of bboxes after nms. Data types supported: S32. |
[in] | max_output_size | An integer tensor representing the maximum number of boxes to be selected by non max suppression. |
[in] | score_threshold | The threshold used to filter detection results. |
[in] | nms_threshold | The threshold used in non maximum suppression. |
Definition at line 32 of file CPPNonMaximumSuppression.cpp.
References ARM_COMPUTE_LOG_PARAMS.
Referenced by CPPDetectionPostProcessLayer::configure().
|
static |
Static function to check if given arguments will lead to a valid configuration of CPPNonMaximumSuppression.
[in] | bboxes | The input bounding boxes tensor info. Data types supported: F32. |
[in] | scores | The corresponding input confidence tensor info. Same as bboxes . |
[out] | indices | The kept indices of bboxes after nms tensor info. Data types supported: S32. |
[in] | max_output_size | An integer tensor representing the maximum number of boxes to be selected by non max suppression. |
[in] | score_threshold | The threshold used to filter detection results. |
[in] | nms_threshold | The threshold used in non maximum suppression. |
Definition at line 46 of file CPPNonMaximumSuppression.cpp.
References CPPNonMaximumSuppressionKernel::validate().
Referenced by CPPDetectionPostProcessLayer::validate().