diff options
| author | SiCong Li <sicong.li@arm.com> | 2019-06-21 16:02:40 +0100 |
|---|---|---|
| committer | James Conroy <james.conroy@arm.com> | 2019-07-09 14:14:41 +0000 |
| commit | c0ed7baa8c05c4710034dfd179fadd31b716a46f (patch) | |
| tree | f8e2a5724d724eab0cfc570a417947c52cb6c451 /tests | |
| parent | 2979457c5974e298033c387a5cd33f95e3301623 (diff) | |
| download | armnn-c0ed7baa8c05c4710034dfd179fadd31b716a46f.tar.gz | |
MLCE-103 Remove hardcoded output shape in ModelAccuracyTool
We can obtain the output tensor shape from the model provided by
the user.
Signed-off-by: SiCong Li <sicong.li@arm.com>
Change-Id: I5074734315174c1b5dc8eea1eff18a4a1c566f2a
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/ModelAccuracyTool-Armnn/ModelAccuracyTool-Armnn.cpp | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/tests/ModelAccuracyTool-Armnn/ModelAccuracyTool-Armnn.cpp b/tests/ModelAccuracyTool-Armnn/ModelAccuracyTool-Armnn.cpp index bb0d824e0e..85241e889c 100644 --- a/tests/ModelAccuracyTool-Armnn/ModelAccuracyTool-Armnn.cpp +++ b/tests/ModelAccuracyTool-Armnn/ModelAccuracyTool-Armnn.cpp @@ -194,6 +194,9 @@ int main(int argc, char* argv[]) inputTensorDataLayout == armnn::DataLayout::NCHW ? inputTensorShape[3] : inputTensorShape[2]; const unsigned int inputTensorHeight = inputTensorDataLayout == armnn::DataLayout::NCHW ? inputTensorShape[2] : inputTensorShape[1]; + // Get output tensor info + const unsigned int outputNumElements = model.GetOutputSize(); + const unsigned int batchSize = 1; // Get normalisation parameters SupportedFrontend modelFrontend; @@ -232,7 +235,7 @@ int main(int argc, char* argv[]) normParams, batchSize, inputTensorDataLayout)); - outputDataContainers = {vector<int>(1001)}; + outputDataContainers = { vector<int>(outputNumElements) }; break; case armnn::DataType::QuantisedAsymm8: inputDataContainers.push_back( @@ -241,7 +244,7 @@ int main(int argc, char* argv[]) normParams, batchSize, inputTensorDataLayout)); - outputDataContainers = {vector<uint8_t>(1001)}; + outputDataContainers = { vector<uint8_t>(outputNumElements) }; break; case armnn::DataType::Float32: default: @@ -251,7 +254,7 @@ int main(int argc, char* argv[]) normParams, batchSize, inputTensorDataLayout)); - outputDataContainers = {vector<float>(1001)}; + outputDataContainers = { vector<float>(outputNumElements) }; break; } |