From a0a0e29f635de08092c2325f8f049ffb286aabaf Mon Sep 17 00:00:00 2001 From: Pablo Tello Date: Fri, 21 Dec 2018 16:47:23 +0000 Subject: COMPMID-1766: Implemented CPP Non Max Suppression Change-Id: I1dcd5fb3d9ad6c6c750415bf8074698b800dfbc1 Reviewed-on: https://review.mlplatform.org/494 Tested-by: Arm Jenkins Reviewed-by: Giuseppe Rossini Reviewed-by: Georgios Pinitas --- tests/datasets/ShapeDatasets.h | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) (limited to 'tests/datasets') diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h index d7ffc127ef..bd29fe649a 100644 --- a/tests/datasets/ShapeDatasets.h +++ b/tests/datasets/ShapeDatasets.h @@ -949,6 +949,37 @@ public: { } }; + +/** Data set containing small 2D tensor shapes. */ +class Small2DNonMaxSuppressionShapes final : public ShapeDataset +{ +public: + Small2DNonMaxSuppressionShapes() + : ShapeDataset("Shape", + { + TensorShape{ 4U, 7U }, + TensorShape{ 4U, 13U }, + TensorShape{ 4U, 64U } + }) + { + } +}; + +/** Data set containing large 2D tensor shapes. */ +class Large2DNonMaxSuppressionShapes final : public ShapeDataset +{ +public: + Large2DNonMaxSuppressionShapes() + : ShapeDataset("Shape", + { + TensorShape{ 4U, 207U }, + TensorShape{ 4U, 113U }, + TensorShape{ 4U, 264U } + }) + { + } +}; + } // namespace datasets } // namespace test } // namespace arm_compute -- cgit v1.2.1