Compute Library
 19.08
CPPBoxWithNonMaximaSuppressionLimit Class Reference

Basic function to run CPPBoxWithNonMaximaSuppressionLimitKernel. More...

#include <CPPBoxWithNonMaximaSuppressionLimit.h>

Collaboration diagram for CPPBoxWithNonMaximaSuppressionLimit:
[legend]

Public Member Functions

void configure (const ITensor *scores_in, const ITensor *boxes_in, const ITensor *batch_splits_in, ITensor *scores_out, ITensor *boxes_out, ITensor *classes, ITensor *batch_splits_out=nullptr, ITensor *keeps=nullptr, ITensor *keeps_size=nullptr, const BoxNMSLimitInfo info=BoxNMSLimitInfo())
 Configure the BoxWithNonMaximaSuppressionLimit CPP kernel. 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...
 

Detailed Description

Basic function to run CPPBoxWithNonMaximaSuppressionLimitKernel.

Definition at line 36 of file CPPBoxWithNonMaximaSuppressionLimit.h.

Member Function Documentation

◆ configure()

void configure ( const ITensor scores_in,
const ITensor boxes_in,
const ITensor batch_splits_in,
ITensor scores_out,
ITensor boxes_out,
ITensor classes,
ITensor batch_splits_out = nullptr,
ITensor keeps = nullptr,
ITensor keeps_size = nullptr,
const BoxNMSLimitInfo  info = BoxNMSLimitInfo() 
)

Configure the BoxWithNonMaximaSuppressionLimit CPP kernel.

Parameters
[in]scores_inThe scores input tensor of size [count, num_classes]. Data types supported: F16/F32
[in]boxes_inThe boxes input tensor of size [count, num_classes * 4]. Data types supported: Same as scores_in
[in]batch_splits_inThe batch splits input tensor of size [batch_size]. Data types supported: Same as scores_in
Note
Can be a nullptr. If not a nullptr, scores_in and boxes_in have items from multiple images.
Parameters
[out]scores_outThe scores output tensor of size [N]. Data types supported: Same as scores_in
[out]boxes_outThe boxes output tensor of size [N, 4]. Data types supported: Same as scores_in
[out]classesThe classes output tensor of size [N]. Data types supported: Same as scores_in
[out]batch_splits_out(Optional) The batch splits output tensor. Data types supported: Same as scores_in
[out]keeps(Optional) The keeps output tensor of size [N]. Data types supported: Same as scores_in
[in]keeps_size(Optional) Number of filtered indices per class tensor of size [num_classes]. Data types supported: Same as scores_in
[in]info(Optional) BoxNMSLimitInfo information.

Definition at line 31 of file CPPBoxWithNonMaximaSuppressionLimit.cpp.

33 {
34  auto k = arm_compute::support::cpp14::make_unique<CPPBoxWithNonMaximaSuppressionLimitKernel>();
35  k->configure(scores_in, boxes_in, batch_splits_in, scores_out, boxes_out, classes, batch_splits_out, keeps, keeps_size, info);
36  _kernel = std::move(k);
37 }

References arm_compute::test::validation::info.


The documentation for this class was generated from the following files: