From 4628d05455dfc179f0437913185e76888115a98a Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Mon, 25 Feb 2019 17:26:05 +0000 Subject: IVGCVSW-2560 Verify Inference test for TensorFlow Lite MobileNet SSD * Assign output shape of MobileNet SSD to ArmNN network * Add m_OverridenOutputShapes to TfLiteParser to set shape in GetNetworkOutputBindingInfo * Use input quantization instead of output quantization params * Correct data and datatype in Inference test Change-Id: I01ac2e07ed08e8928ba0df33a4847399e1dd8394 Signed-off-by: Narumol Prangnawarat Signed-off-by: Aron Virginas-Tar --- include/armnn/Descriptors.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'include') diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index b14ed0b1ab..dc09cc60e8 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -332,7 +332,7 @@ struct DetectionPostProcessDescriptor DetectionPostProcessDescriptor() : m_MaxDetections(0) , m_MaxClassesPerDetection(1) - , m_DetectionsPerClass(100) + , m_DetectionsPerClass(1) , m_NmsScoreThreshold(0) , m_NmsIouThreshold(0) , m_NumClasses(0) -- cgit v1.2.1