aboutsummaryrefslogtreecommitdiff
path: root/include
diff options
context:
space:
mode:
authorNarumol Prangnawarat <narumol.prangnawarat@arm.com>2019-01-25 10:46:40 +0000
committerNarumol Prangnawarat <narumol.prangnawarat@arm.com>2019-01-25 16:59:16 +0000
commita0d56c77a53f5f991565041927003ce7460730ce (patch)
treede08bfa6f95752ec8e7b5b58d0b689288f44b36a /include
parent12055747d47657a89d60748a078897f6436e6aa0 (diff)
downloadarmnn-a0d56c77a53f5f991565041927003ce7460730ce.tar.gz
IVGCVSW-2556 Add Layer implementation for Detection PostProcess
* Add DetectionPostProcessDescriptor. * Add implementation for DetectionPostProcessLayer. * Unit test to validate output. Change-Id: If63e83eb2a2978c549071c7aeb272906e7c35fe9
Diffstat (limited to 'include')
-rw-r--r--include/armnn/Descriptors.hpp36
1 files changed, 36 insertions, 0 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp
index 44235c7ada..29d294e69f 100644
--- a/include/armnn/Descriptors.hpp
+++ b/include/armnn/Descriptors.hpp
@@ -329,6 +329,42 @@ struct DepthwiseConvolution2dDescriptor
struct DetectionPostProcessDescriptor
{
+ DetectionPostProcessDescriptor()
+ : m_MaxDetections(0)
+ , m_MaxClassesPerDetection(1)
+ , m_DetectionsPerClass(100)
+ , m_NmsScoreThreshold(0)
+ , m_NmsIouThreshold(0)
+ , m_NumClasses(0)
+ , m_UseRegularNms(false)
+ , m_ScaleX(0)
+ , m_ScaleY(0)
+ , m_ScaleW(0)
+ , m_ScaleH(0)
+ {}
+
+ /// Maximum numbers of detections.
+ uint32_t m_MaxDetections;
+ /// Maximum numbers of classes per detection, used in Fast NMS.
+ uint32_t m_MaxClassesPerDetection;
+ /// Detections per classes, used in Regular NMS.
+ uint32_t m_DetectionsPerClass;
+ /// NMS score threshold.
+ float m_NmsScoreThreshold;
+ /// Intersection over union threshold.
+ float m_NmsIouThreshold;
+ /// Number of classes.
+ int32_t m_NumClasses;
+ /// Use Regular NMS.
+ bool m_UseRegularNms;
+ /// Center size encoding scale x.
+ float m_ScaleX;
+ /// Center size encoding scale y.
+ float m_ScaleY;
+ /// Center size encoding scale weight.
+ float m_ScaleW;
+ /// Center size encoding scale height.
+ float m_ScaleH;
};
/// A NormalizationDescriptor for the NormalizationLayer.