24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_NEGENERATEPROPOSALSLAYER_H
25 #define ARM_COMPUTE_NEGENERATEPROPOSALSLAYER_H
43 class NEComputeAllAnchorsKernel;
140 std::unique_ptr<NEComputeAllAnchorsKernel> _compute_anchors;
156 Tensor _deltas_flattened_f32;
162 Tensor _all_proposals_quantized;
164 Tensor _classes_nms_unused;
165 Tensor _proposals_4_roi_values;
168 Tensor *_all_proposals_to_use;
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...
Basic function to run cpu::kernels::CpuReshapeKernel.
Base class for all functions.
~NEGenerateProposalsLayer()
Default destructor.
Interface for CPU tensor.
Basic function to run CPPBoxWithNonMaximaSuppressionLimitKernel.
Basic function to generate proposals for a RPN (Region Proposal Network)
NEGenerateProposalsLayer & operator=(const NEGenerateProposalsLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
Basic function to run a quantization layer using cpu::CpuQuantize.
Basic function to pad a tensor.
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 function to run cpu::kernels::CpuPermuteKernel.
Basic pool of threads to execute CPP/Neon code on several cores in parallel.
NEGenerateProposalsLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Default constructor.
Copyright (c) 2017-2024 Arm Limited.
Basic function to run cpu::CpuDequantize that dequantizes an input tensor.
Store the tensor's metadata.
Basic implementation of the tensor interface.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
Generate Proposals Information class.
void run() override
Run the kernels contained in the function.