32 const float score_threshold,
const float nms_threshold)
34 auto k = std::make_unique<CPPNonMaximumSuppressionKernel>();
35 k->configure(bboxes, scores, indices, max_output_size, score_threshold, nms_threshold);
36 _kernel = std::move(k);
41 const float score_threshold,
const float nms_threshold)
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...
Store the tensor's metadata.
Interface for Neon tensor.
Copyright (c) 2017-2021 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...
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.