diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-03-18 12:37:06 +0000 |
---|---|---|
committer | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-03-18 12:37:06 +0000 |
commit | 3e14a9d2033530df49546ab0da63ad4b6470f551 (patch) | |
tree | 030433752c31b42eeecc1ed094c39c2c508dd57f /src/armnnSerializer/ArmnnSchema.fbs | |
parent | 65d30965eef7e8534fc16ea4ded413c42a81c362 (diff) | |
download | armnn-3e14a9d2033530df49546ab0da63ad4b6470f551.tar.gz |
IVGCVSW-2686 Add Serializer and Deserializer for DetectionPostProcess
Change-Id: Ife48db5fdb005ebca0a6f21862b0ce971ccf58b7
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Diffstat (limited to 'src/armnnSerializer/ArmnnSchema.fbs')
-rw-r--r-- | src/armnnSerializer/ArmnnSchema.fbs | 26 |
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 { |