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/core/CPP/CPPKernels.h | 1 + .../CPPBoxWithNonMaximaSuppressionLimitKernel.h | 91 ++++++++++++++++++++++ 2 files changed, 92 insertions(+) create mode 100644 arm_compute/core/CPP/kernels/CPPBoxWithNonMaximaSuppressionLimitKernel.h (limited to 'arm_compute/core/CPP') diff --git a/arm_compute/core/CPP/CPPKernels.h b/arm_compute/core/CPP/CPPKernels.h index bf24a94aa7..39b77cde45 100644 --- a/arm_compute/core/CPP/CPPKernels.h +++ b/arm_compute/core/CPP/CPPKernels.h @@ -25,6 +25,7 @@ #define __ARM_COMPUTE_CPPKERNELS_H__ /* Header regrouping all the CPP kernels */ +#include "arm_compute/core/CPP/kernels/CPPBoxWithNonMaximaSuppressionLimitKernel.h" #include "arm_compute/core/CPP/kernels/CPPCornerCandidatesKernel.h" #include "arm_compute/core/CPP/kernels/CPPDetectionWindowNonMaximaSuppressionKernel.h" #include "arm_compute/core/CPP/kernels/CPPFlipWeightsKernel.h" diff --git a/arm_compute/core/CPP/kernels/CPPBoxWithNonMaximaSuppressionLimitKernel.h b/arm_compute/core/CPP/kernels/CPPBoxWithNonMaximaSuppressionLimitKernel.h new file mode 100644 index 0000000000..a1adca3a83 --- /dev/null +++ b/arm_compute/core/CPP/kernels/CPPBoxWithNonMaximaSuppressionLimitKernel.h @@ -0,0 +1,91 @@ +/* + * 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_CPPBOXWITHNONMAXIMASUPPRESSIONLIMITKERNEL_H__ +#define __ARM_COMPUTE_CPPBOXWITHNONMAXIMASUPPRESSIONLIMITKERNEL_H__ + +#include "arm_compute/core/IArray.h" +#include "arm_compute/core/IHOG.h" +#include "arm_compute/core/NEON/INEKernel.h" +#include "arm_compute/core/Types.h" + +namespace arm_compute +{ +class ITensor; + +/** CPP kernel to perform computation of BoxWithNonMaximaSuppressionLimit */ +class CPPBoxWithNonMaximaSuppressionLimitKernel : public ICPPKernel +{ +public: + const char *name() const override + { + return "CPPBoxWithNonMaximaSuppressionLimitKernel"; + } + /** Default constructor */ + CPPBoxWithNonMaximaSuppressionLimitKernel(); + /** Prevent instances of this class from being copied (As this class contains pointers) */ + CPPBoxWithNonMaximaSuppressionLimitKernel(const CPPBoxWithNonMaximaSuppressionLimitKernel &) = delete; + /** Prevent instances of this class from being copied (As this class contains pointers) */ + CPPBoxWithNonMaximaSuppressionLimitKernel &operator=(const CPPBoxWithNonMaximaSuppressionLimitKernel &) = delete; + /** Allow instances of this class to be moved */ + CPPBoxWithNonMaximaSuppressionLimitKernel(CPPBoxWithNonMaximaSuppressionLimitKernel &&) = default; + /** Allow instances of this class to be moved */ + CPPBoxWithNonMaximaSuppressionLimitKernel &operator=(CPPBoxWithNonMaximaSuppressionLimitKernel &&) = default; + /** Initialise the kernel's input and output tensors. + * + * @param[in] scores_in The scores input tensor of size [num_classes, count]. Data types supported: F16/F32 + * @param[in] boxes_in The boxes input tensor of size [num_classes * 4, count]. 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 [4, N]. 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 [batch_size]. 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[out] 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()); + + // Inherited methods overridden: + void run(const Window &window, const ThreadInfo &info) override; + bool is_parallelisable() const override; + + template + void run_nmslimit(); + +private: + 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; + ITensor *_keeps; + ITensor *_keeps_size; + BoxNMSLimitInfo _info; +}; +} // namespace arm_compute +#endif /* __ARM_COMPUTE_CPPBOXWITHNONMAXIMASUPPRESSIONLIMITKERNEL_H__ */ -- cgit v1.2.1