diff options
author | Pablo Tello <pablo.tello@arm.com> | 2018-12-21 16:47:23 +0000 |
---|---|---|
committer | VidhyaSudhan Loganathan <vidhyasudhan.loganathan@arm.com> | 2019-02-15 14:22:15 +0000 |
commit | e96e4f0e49a3571f823927913dde173513baee72 (patch) | |
tree | 773188bbe1add7f62cbc982f7bb68be7e8cc2916 /tests/datasets | |
parent | bb365de1e14144f239f03de00db9b41f61bf7373 (diff) | |
download | ComputeLibrary-e96e4f0e49a3571f823927913dde173513baee72.tar.gz |
COMPMID-1766: Implemented CPP Non Max Suppression
Change-Id: I2d2b684d464f7b3bb1f91cfd29952f612d65f11f
Signed-off-by: Pablo Tello <pablo.tello@arm.com>
Reviewed-on: https://review.mlplatform.org/708
Reviewed-by: VidhyaSudhan Loganathan <vidhyasudhan.loganathan@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/datasets')
-rw-r--r-- | tests/datasets/ShapeDatasets.h | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h index 480df3c7eb..f461d7f9d2 100644 --- a/tests/datasets/ShapeDatasets.h +++ b/tests/datasets/ShapeDatasets.h @@ -946,6 +946,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 |