IVGCVSW-2686 Add Serializer and Deserializer for DetectionPostProcess

Change-Id: Ife48db5fdb005ebca0a6f21862b0ce971ccf58b7
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>

1 files changed, 24 insertions, 2 deletions

diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index 40ee7a5e45..00c1a4502b 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -113,7 +113,8 @@ enum LayerType : uint {
     Mean = 29,
     Merger = 30,
     L2Normalization = 31,
-    Splitter = 32
+    Splitter = 32,
+    DetectionPostProcess = 33
 }
 
 // Base layer table to be used as part of other layers
@@ -453,6 +454,26 @@ table SplitterLayer {
    descriptor:ViewsDescriptor;
 }
 
+table DetectionPostProcessLayer {
+    base:LayerBase;
+    descriptor:DetectionPostProcessDescriptor;
+    anchors:ConstTensor;
+}
+
+table DetectionPostProcessDescriptor {
+    maxDetections:uint;
+    maxClassesPerDetection:uint;
+    detectionsPerClass:uint;
+    nmsScoreThreshold:float;
+    nmsIouThreshold:float;
+    numClasses:uint;
+    useRegularNms:bool;
+    scaleX:float;
+    scaleY:float;
+    scaleW:float;
+    scaleH:float;
+}
+
 union Layer {
     ActivationLayer,
     AdditionLayer,
@@ -486,7 +507,8 @@ union Layer {
     MeanLayer,
     MergerLayer,
     L2NormalizationLayer,
-    SplitterLayer
+    SplitterLayer,
+    DetectionPostProcessLayer
 }
 
 table AnyLayer {