From 2709d61ec25156387e48a75d19a5ab5a035fc12b Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Wed, 19 Sep 2018 09:46:47 +0100 Subject: COMPMID-1328 Add support for BoxWithNMSLimit operator in CPP Change-Id: I5aae537372bf797fbb2a2bae81038f8963b041a9 --- arm_compute/runtime/CPP/CPPFunctions.h | 1 + .../CPPBoxWithNonMaximaSuppressionLimit.h | 57 ++++++++++++++++++++++ 2 files changed, 58 insertions(+) create mode 100644 arm_compute/runtime/CPP/functions/CPPBoxWithNonMaximaSuppressionLimit.h (limited to 'arm_compute/runtime/CPP') diff --git a/arm_compute/runtime/CPP/CPPFunctions.h b/arm_compute/runtime/CPP/CPPFunctions.h index 4b7e1afbb5..9c17d9e404 100644 --- a/arm_compute/runtime/CPP/CPPFunctions.h +++ b/arm_compute/runtime/CPP/CPPFunctions.h @@ -25,6 +25,7 @@ #define __ARM_COMPUTE_CPPFUNCTIONS_H__ /* Header regrouping all the CPP functions */ +#include "arm_compute/runtime/CPP/functions/CPPBoxWithNonMaximaSuppressionLimit.h" #include "arm_compute/runtime/CPP/functions/CPPPermute.h" #include "arm_compute/runtime/CPP/functions/CPPUpsample.h" diff --git a/arm_compute/runtime/CPP/functions/CPPBoxWithNonMaximaSuppressionLimit.h b/arm_compute/runtime/CPP/functions/CPPBoxWithNonMaximaSuppressionLimit.h new file mode 100644 index 0000000000..8a886a42cd --- /dev/null +++ b/arm_compute/runtime/CPP/functions/CPPBoxWithNonMaximaSuppressionLimit.h @@ -0,0 +1,57 @@ +/* + * Copyright (c) 2018 ARM Limited. + * + * SPDX-License-Identifier: MIT + * + * Permission is hereby granted, free of charge, to any person obtaining a copy + * of this software and associated documentation files (the "Software"), to + * deal in the Software without restriction, including without limitation the + * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or + * sell copies of the Software, and to permit persons to whom the Software is + * furnished to do so, subject to the following conditions: + * + * The above copyright notice and this permission notice shall be included in all + * copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR + * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, + * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE + * SOFTWARE. + */ +#ifndef __ARM_COMPUTE_CPPBOXWITHNONMAXIMASUPPRESSIONLIMIT_H__ +#define __ARM_COMPUTE_CPPBOXWITHNONMAXIMASUPPRESSIONLIMIT_H__ + +#include "arm_compute/runtime/CPP/ICPPSimpleFunction.h" + +#include "arm_compute/core/Types.h" + +namespace arm_compute +{ +class ITensor; + +/** Basic function to run @ref CPPBoxWithNonMaximaSuppressionLimitKernel */ +class CPPBoxWithNonMaximaSuppressionLimit : public ICPPSimpleFunction +{ +public: + /** Configure the BoxWithNonMaximaSuppressionLimit CPP kernel + * + * @param[in] scores_in The scores input tensor of size [count, num_classes]. Data types supported: F16/F32 + * @param[in] boxes_in The boxes input tensor of size [count, num_classes * 4]. Data types supported: Same as @p scores_in + * @param[in] batch_splits_in The batch splits input tensor of size [batch_size]. Data types supported: Same as @p scores_in + * @note Can be a nullptr. If not a nullptr, @p scores_in and @p boxes_in have items from multiple images. + * @param[out] scores_out The scores output tensor of size [N]. Data types supported: Same as @p scores_in + * @param[out] boxes_out The boxes output tensor of size [N, 4]. Data types supported: Same as @p scores_in + * @param[out] classes The classes output tensor of size [N]. Data types supported: Same as @p scores_in + * @param[out] batch_splits_out (Optional) The batch splits output tensor. Data types supported: Same as @p scores_in + * @param[out] keeps (Optional) The keeps output tensor of size [N]. Data types supported: Same as @p scores_in + * @param[in] keeps_size (Optional) Number of filtered indices per class tensor of size [num_classes]. Data types supported: Same as @p scores_in + * @param[in] info (Optional) BoxNMSLimitInfo information. + */ + void configure(const ITensor *scores_in, const ITensor *boxes_in, const ITensor *batch_splits_in, ITensor *scores_out, ITensor *boxes_out, ITensor *classes, + ITensor *batch_splits_out = nullptr, ITensor *keeps = nullptr, ITensor *keeps_size = nullptr, const BoxNMSLimitInfo info = BoxNMSLimitInfo()); +}; +} // namespace arm_compute +#endif /* __ARM_COMPUTE_CPPBOXWITHNONMAXIMASUPPRESSIONLIMIT_H__ */ -- cgit v1.2.1