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>
diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h
index 480df3c..f461d7f 100644
--- a/tests/datasets/ShapeDatasets.h
+++ b/tests/datasets/ShapeDatasets.h
@@ -946,6 +946,37 @@
{
}
};
+
+/** 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