From 4fcda0101ec3d110c1d6d7bee5c83416b645528a Mon Sep 17 00:00:00 2001 From: telsoa01 Date: Fri, 9 Mar 2018 14:13:49 +0000 Subject: Release 18.02 Change-Id: Id3c11dc5ee94ef664374a988fcc6901e9a232fa6 --- tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp | 39 +++++++++++++++++++++++++++++++ 1 file changed, 39 insertions(+) create mode 100644 tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp (limited to 'tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp') diff --git a/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp b/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp new file mode 100644 index 0000000000..af60be95ec --- /dev/null +++ b/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp @@ -0,0 +1,39 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// See LICENSE file in the project root for full license information. +// +#include "../YoloInferenceTest.hpp" +#include "armnnCaffeParser/ICaffeParser.hpp" +#include "armnn/TypesUtils.hpp" + +int main(int argc, char* argv[]) +{ + armnn::TensorShape inputTensorShape{ { 1, 3, YoloImageHeight, YoloImageWidth } }; + + using YoloInferenceModel = InferenceModel; + + return InferenceTestMain(argc, argv, { 0 }, + [&inputTensorShape]() + { + return make_unique>( + [&] + (typename YoloInferenceModel::CommandLineOptions modelOptions) + { + if (!ValidateDirectory(modelOptions.m_ModelDir)) + { + return std::unique_ptr(); + } + + typename YoloInferenceModel::Params modelParams; + modelParams.m_ModelPath = modelOptions.m_ModelDir + "yolov1_tiny_voc2007_model.caffemodel"; + modelParams.m_InputBinding = "data"; + modelParams.m_OutputBinding = "fc12"; + modelParams.m_InputTensorShape = &inputTensorShape; + modelParams.m_IsModelBinary = true; + modelParams.m_ComputeDevice = modelOptions.m_ComputeDevice; + + return std::make_unique(modelParams); + }); + }); +} -- cgit v1.2.1