Release 18.02
Change-Id: Id3c11dc5ee94ef664374a988fcc6901e9a232fa6
diff --git a/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp b/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp
new file mode 100644
index 0000000..af60be9
--- /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<armnnCaffeParser::ICaffeParser,
+ float>;
+
+ return InferenceTestMain(argc, argv, { 0 },
+ [&inputTensorShape]()
+ {
+ return make_unique<YoloTestCaseProvider<YoloInferenceModel>>(
+ [&]
+ (typename YoloInferenceModel::CommandLineOptions modelOptions)
+ {
+ if (!ValidateDirectory(modelOptions.m_ModelDir))
+ {
+ return std::unique_ptr<YoloInferenceModel>();
+ }
+
+ 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<YoloInferenceModel>(modelParams);
+ });
+ });
+}