aboutsummaryrefslogtreecommitdiff
path: root/src/armnnTfLiteParser/test/DetectionPostProcess.cpp
diff options
context:
space:
mode:
authorNarumol Prangnawarat <narumol.prangnawarat@arm.com>2019-02-25 17:26:05 +0000
committerNarumol Prangnawarat <narumol.prangnawarat@arm.com>2019-02-26 17:41:15 +0000
commit4628d05455dfc179f0437913185e76888115a98a (patch)
treea8eac68ee5aee88a7071ac6f13af7932b98caa87 /src/armnnTfLiteParser/test/DetectionPostProcess.cpp
parent452869973b9a45c9c44820d16f92f7dfc96e9aef (diff)
downloadarmnn-4628d05455dfc179f0437913185e76888115a98a.tar.gz
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 <narumol.prangnawarat@arm.com> Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Diffstat (limited to 'src/armnnTfLiteParser/test/DetectionPostProcess.cpp')
-rw-r--r--src/armnnTfLiteParser/test/DetectionPostProcess.cpp4
1 files changed, 0 insertions, 4 deletions
diff --git a/src/armnnTfLiteParser/test/DetectionPostProcess.cpp b/src/armnnTfLiteParser/test/DetectionPostProcess.cpp
index 3c602937ce..3002885016 100644
--- a/src/armnnTfLiteParser/test/DetectionPostProcess.cpp
+++ b/src/armnnTfLiteParser/test/DetectionPostProcess.cpp
@@ -76,28 +76,24 @@ struct DetectionPostProcessFixture : ParserFlatbuffersFixture
}
},
{
- "shape": [1, 3, 4],
"type": "FLOAT32",
"buffer": 3,
"name": "detection_boxes",
"quantization": {}
},
{
- "shape": [1, 3],
"type": "FLOAT32",
"buffer": 4,
"name": "detection_classes",
"quantization": {}
},
{
- "shape": [1, 3],
"type": "FLOAT32",
"buffer": 5,
"name": "detection_scores",
"quantization": {}
},
{
- "shape": [1],
"type": "FLOAT32",
"buffer": 6,
"name": "num_detections",