24 #ifndef ARM_COMPUTE_CPP_DETECTION_OUTPUT_LAYER_H 25 #define ARM_COMPUTE_CPP_DETECTION_OUTPUT_LAYER_H 85 std::vector<LabelBBox> _all_location_predictions;
86 std::vector<std::map<int, std::vector<float>>> _all_confidence_scores;
87 std::vector<BBox> _all_prior_bboxes;
88 std::vector<std::array<float, 4>> _all_prior_variances;
89 std::vector<LabelBBox> _all_decode_bboxes;
90 std::vector<std::map<int, std::vector<int>>> _all_indices;
CPPDetectionOutputLayer()
Default constructor.
Base class for all functions.
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.
CPP Function to generate the detection output based on location and confidence predictions by doing n...
Store the tensor's metadata.
Interface for Neon tensor.
void run() override
Run the kernels contained in the function.
Copyright (c) 2017-2021 Arm Limited.
Detection Output layer info.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
CPPDetectionOutputLayer & operator=(const CPPDetectionOutputLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
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...