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;
154 std::unique_ptr<CLComputeAllAnchorsKernel> _compute_anchors_kernel;
155 std::unique_ptr<CLBoundingBoxTransformKernel> _bounding_box_kernel;
156 std::unique_ptr<CLPadLayerKernel> _pad_kernel;
157 std::unique_ptr<CLDequantizationLayer> _dequantize_anchors;
158 std::unique_ptr<CLDequantizationLayer> _dequantize_deltas;
159 std::unique_ptr<CLQuantizationLayer> _quantize_all_proposals;
189 void run_cpp_nms_kernel();
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.
Generate Proposals Information class.
Base class for all functions.
void run() override
Run the kernels contained in the function.
Store the tensor's metadata.
Copyright (c) 2017-2021 Arm Limited.
Interface to enqueue OpenCL kernels and get/set the OpenCL CommandQueue and ICLTuner.
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...
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
Basic function to run CPPBoxWithNonMaximaSuppressionLimitKernel.
Interface for OpenCL tensor.
CLGenerateProposalsLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Default constructor.
Basic function to run opencl::kernels::ClReshapeKernel.
Basic function to generate proposals for a RPN (Region Proposal Network)
~CLGenerateProposalsLayer()
Default destructor.
Basic implementation of the OpenCL tensor interface.