21.02
|
CPP Function to generate the detection output based on location and confidence predictions by doing non maximum suppression. More...
#include <CPPDetectionOutputLayer.h>
Public Member Functions | |
CPPDetectionOutputLayer () | |
Default constructor. More... | |
void | configure (const ITensor *input_loc, const ITensor *input_conf, const ITensor *input_priorbox, ITensor *output, DetectionOutputLayerInfo info=DetectionOutputLayerInfo()) |
Configure the detection output layer CPP kernel. More... | |
void | run () override |
Run the kernels contained in the function. More... | |
CPPDetectionOutputLayer (const CPPDetectionOutputLayer &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
CPPDetectionOutputLayer & | operator= (const CPPDetectionOutputLayer &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) 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 *input_loc, const ITensorInfo *input_conf, const ITensorInfo *input_priorbox, const ITensorInfo *output, DetectionOutputLayerInfo info=DetectionOutputLayerInfo()) |
Static function to check if given info will lead to a valid configuration of CPPDetectionOutputLayer. More... | |
CPP Function to generate the detection output based on location and confidence predictions by doing non maximum suppression.
Definition at line 40 of file CPPDetectionOutputLayer.h.
Default constructor.
Definition at line 385 of file CPPDetectionOutputLayer.cpp.
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
void configure | ( | const ITensor * | input_loc, |
const ITensor * | input_conf, | ||
const ITensor * | input_priorbox, | ||
ITensor * | output, | ||
DetectionOutputLayerInfo | info = DetectionOutputLayerInfo() |
||
) |
Configure the detection output layer CPP kernel.
[in] | input_loc | The mbox location input tensor of size [C1, N]. Data types supported: F32. |
[in] | input_conf | The mbox confidence input tensor of size [C2, N]. Data types supported: F32. |
[in] | input_priorbox | The mbox prior box input tensor of size [C3, 2, N]. Data types supported: F32. |
[out] | output | The output tensor of size [7, M]. Data types supported: Same as input |
[in] | info | (Optional) DetectionOutputLayerInfo information. |
Definition at line 391 of file CPPDetectionOutputLayer.cpp.
References ARM_COMPUTE_ERROR_ON_NULLPTR, ARM_COMPUTE_ERROR_THROW_ON, arm_compute::auto_init_if_empty(), DetectionOutputLayerInfo::background_label_id(), ICloneable< T >::clone(), ITensorInfo::dimension(), ITensor::info(), arm_compute::test::validation::info, DetectionOutputLayerInfo::keep_top_k(), ITensorInfo::num_dimensions(), DetectionOutputLayerInfo::num_loc_classes(), Dimensions< T >::set_num_dimensions(), ITensorInfo::set_valid_region(), DetectionOutputLayerInfo::share_location(), ITensorInfo::tensor_shape(), arm_compute::U, and arm_compute::validate_arguments().
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
|
overridevirtual |
Run the kernels contained in the function.
For Neon kernels:
For OpenCL kernels:
Implements IFunction.
Definition at line 444 of file CPPDetectionOutputLayer.cpp.
References ARM_COMPUTE_ERROR_ON, ARM_COMPUTE_ERROR_ON_MSG_VAR, ARM_COMPUTE_ERROR_VAR, DetectionOutputLayerInfo::background_label_id(), DetectionOutputLayerInfo::code_type(), DetectionOutputLayerInfo::confidence_threshold(), arm_compute::mlgo::parser::end(), DetectionOutputLayerInfo::eta(), enable_tracing::find(), ITensor::info(), DetectionOutputLayerInfo::keep_top_k(), DetectionOutputLayerInfo::nms_threshold(), DetectionOutputLayerInfo::num_classes(), DetectionOutputLayerInfo::num_loc_classes(), ITensor::ptr_to_element(), ITensorInfo::set_valid_region(), DetectionOutputLayerInfo::share_location(), DetectionOutputLayerInfo::top_k(), and DetectionOutputLayerInfo::variance_encoded_in_target().
|
static |
Static function to check if given info will lead to a valid configuration of CPPDetectionOutputLayer.
[in] | input_loc | The mbox location input tensor info. Data types supported: F32. |
[in] | input_conf | The mbox confidence input tensor info. Data types supported: F32. |
[in] | input_priorbox | The mbox prior box input tensor info. Data types supported: F32. |
[in] | output | The output tensor info. Data types supported: Same as input |
[in] | info | (Optional) DetectionOutputLayerInfo information. |
Definition at line 438 of file CPPDetectionOutputLayer.cpp.
References ARM_COMPUTE_RETURN_ON_ERROR, and arm_compute::validate_arguments().