diff options
Diffstat (limited to 'src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp')
-rw-r--r-- | src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp | 37 |
1 files changed, 34 insertions, 3 deletions
diff --git a/src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp b/src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp index 50e674ef2c..9eb2f2b93d 100644 --- a/src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp +++ b/src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp @@ -5,13 +5,16 @@ #pragma once +#include <armnn/Descriptors.hpp> +#include <armnn/IRuntime.hpp> +#include <armnn/TypesUtils.hpp> + #include "Schema.hpp" #include <boost/filesystem.hpp> #include <boost/assert.hpp> #include <boost/format.hpp> #include <experimental/filesystem> -#include <armnn/IRuntime.hpp> -#include <armnn/TypesUtils.hpp> + #include "test/TensorHelpers.hpp" #include "TypeUtils.hpp" @@ -21,6 +24,7 @@ #include "flatbuffers/idl.h" #include "flatbuffers/util.h" +#include "flatbuffers/flexbuffers.h" #include <schema_generated.h> #include <iostream> @@ -159,6 +163,33 @@ struct ParserFlatbuffersFixture const std::map<std::string, std::vector<DataType1>>& inputData, const std::map<std::string, std::vector<DataType2>>& expectedOutputData); + static inline std::string GenerateDetectionPostProcessJsonString( + const armnn::DetectionPostProcessDescriptor& descriptor) + { + flexbuffers::Builder detectPostProcess; + detectPostProcess.Map([&]() { + detectPostProcess.Bool("use_regular_nms", descriptor.m_UseRegularNms); + detectPostProcess.Int("max_detections", descriptor.m_MaxDetections); + detectPostProcess.Int("max_classes_per_detection", descriptor.m_MaxClassesPerDetection); + detectPostProcess.Int("detections_per_class", descriptor.m_DetectionsPerClass); + detectPostProcess.Int("num_classes", descriptor.m_NumClasses); + detectPostProcess.Float("nms_score_threshold", descriptor.m_NmsScoreThreshold); + detectPostProcess.Float("nms_iou_threshold", descriptor.m_NmsIouThreshold); + detectPostProcess.Float("h_scale", descriptor.m_ScaleH); + detectPostProcess.Float("w_scale", descriptor.m_ScaleW); + detectPostProcess.Float("x_scale", descriptor.m_ScaleX); + detectPostProcess.Float("y_scale", descriptor.m_ScaleY); + }); + detectPostProcess.Finish(); + + // Create JSON string + std::stringstream strStream; + std::vector<uint8_t> buffer = detectPostProcess.GetBuffer(); + std::copy(buffer.begin(), buffer.end(),std::ostream_iterator<int>(strStream,",")); + + return strStream.str(); + } + void CheckTensors(const TensorRawPtr& tensors, size_t shapeSize, const std::vector<int32_t>& shape, tflite::TensorType tensorType, uint32_t buffer, const std::string& name, const std::vector<float>& min, const std::vector<float>& max, @@ -310,4 +341,4 @@ void ParserFlatbuffersFixture::RunTest(std::size_t subgraphId, } } } -} +}
\ No newline at end of file |