24.02.1
|
Go to the documentation of this file.
35 unsigned int max_output_size,
36 const float score_threshold,
37 const float nms_threshold)
41 auto k = std::make_unique<CPPNonMaximumSuppressionKernel>();
42 k->configure(bboxes, scores, indices, max_output_size, score_threshold, nms_threshold);
43 _kernel = std::move(k);
49 unsigned int max_output_size,
50 const float score_threshold,
51 const float nms_threshold)
Interface for CPU tensor.
static Status validate(const ITensorInfo *input_bboxes, const ITensorInfo *input_scores, const ITensorInfo *output_indices, unsigned int max_output_size, const float score_threshold, const float iou_threshold)
Static function to check if given arguments will lead to a valid configuration of CPPNonMaximumSuppre...
Copyright (c) 2017-2024 Arm Limited.
static Status validate(const ITensorInfo *bboxes, const ITensorInfo *scores, const ITensorInfo *indices, unsigned int max_output_size, const float score_threshold, const float nms_threshold)
Static function to check if given arguments will lead to a valid configuration of CPPNonMaximumSuppre...
Store the tensor's metadata.
#define ARM_COMPUTE_LOG_PARAMS(...)
void configure(const ITensor *bboxes, const ITensor *scores, ITensor *indices, unsigned int max_output_size, const float score_threshold, const float nms_threshold)
Configure the function to perform non maximal suppression.