24 #ifndef ARM_COMPUTE_NEGENERATEPROPOSALSLAYER_H 25 #define ARM_COMPUTE_NEGENERATEPROPOSALSLAYER_H 43 class NEComputeAllAnchorsKernel;
121 std::unique_ptr<NEComputeAllAnchorsKernel> _compute_anchors;
137 Tensor _deltas_flattened_f32;
143 Tensor _all_proposals_quantized;
145 Tensor _classes_nms_unused;
146 Tensor _proposals_4_roi_values;
149 Tensor *_all_proposals_to_use;
Generate Proposals Information class.
~NEGenerateProposalsLayer()
Default destructor.
Base class for all functions.
void run() override
Run the kernels contained in the function.
Basic function to generate proposals for a RPN (Region Proposal Network)
Store the tensor's metadata.
Basic function to simulate a quantization layer.
Basic function to run cpu::kernels::CpuPermuteKernel.
Interface for Neon tensor.
NEGenerateProposalsLayer & operator=(const NEGenerateProposalsLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
static Status validate(const ITensorInfo *scores, const ITensorInfo *deltas, const ITensorInfo *anchors, const ITensorInfo *proposals, const ITensorInfo *scores_out, const ITensorInfo *num_valid_proposals, const GenerateProposalsInfo &info)
Static function to check if given info will lead to a valid configuration of NEGenerateProposalsLayer...
Copyright (c) 2017-2021 Arm Limited.
void configure(const ITensor *scores, const ITensor *deltas, const ITensor *anchors, ITensor *proposals, ITensor *scores_out, ITensor *num_valid_proposals, const GenerateProposalsInfo &info)
Set the input and output tensors.
Basic implementation of the tensor interface.
Basic function to run NEDequantizationLayerKernel that dequantizes an input tensor.
NEGenerateProposalsLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Default constructor.
Basic pool of threads to execute CPP/Neon code on several cores in parallel.
Basic function to run cpu::kernels::CpuReshapeKernel.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
Basic function to run CPPBoxWithNonMaximaSuppressionLimitKernel.
Basic function to pad a tensor.