24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CLGENERATEPROPOSALSLAYER_H
25 #define ARM_COMPUTE_CLGENERATEPROPOSALSLAYER_H
41 class CLCompileContext;
42 class CLBoundingBoxTransformKernel;
43 class CLDequantizationLayer;
44 class CLComputeAllAnchorsKernel;
45 class CLPadLayerKernel;
46 class CLQuantizationLayer;
169 std::unique_ptr<CLComputeAllAnchorsKernel> _compute_anchors_kernel;
170 std::unique_ptr<CLBoundingBoxTransformKernel> _bounding_box_kernel;
171 std::unique_ptr<CLPadLayerKernel> _pad_kernel;
172 std::unique_ptr<CLDequantizationLayer> _dequantize_anchors;
173 std::unique_ptr<CLDequantizationLayer> _dequantize_deltas;
174 std::unique_ptr<CLQuantizationLayer> _quantize_all_proposals;
204 void run_cpp_nms_kernel();
CLGenerateProposalsLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Default constructor.
Base class for all functions.
void configure(const ICLTensor *scores, const ICLTensor *deltas, const ICLTensor *anchors, ICLTensor *proposals, ICLTensor *scores_out, ICLTensor *num_valid_proposals, const GenerateProposalsInfo &info)
Set the input and output tensors.
void run() override
Run the kernels contained in the function.
Interface for OpenCL tensor.
Basic function to run CPPBoxWithNonMaximaSuppressionLimitKernel.
Basic implementation of the OpenCL tensor interface.
Interface to enqueue OpenCL kernels and get/set the OpenCL CommandQueue and ICLTuner.
Basic function to run opencl::kernels::ClReshapeKernel.
CLGenerateProposalsLayer & operator=(const CLGenerateProposalsLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
Basic function to execute an opencl::kernels::ClPermuteKernel.
Basic pool of threads to execute CPP/Neon code on several cores in parallel.
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 CLGenerateProposalsLayer...
Copyright (c) 2017-2024 Arm Limited.
Basic function to generate proposals for a RPN (Region Proposal Network)
Store the tensor's metadata.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
~CLGenerateProposalsLayer()
Default destructor.
Generate Proposals Information class.