diff options
author | Nikhil Raj <nikhil.raj@arm.com> | 2021-04-02 22:04:39 +0100 |
---|---|---|
committer | Nikhil Raj <nikhil.raj@arm.com> | 2021-04-16 14:27:27 +0100 |
commit | 6dd178f2395b34cfb360eabb0130c19ed258f5fa (patch) | |
tree | 6814e4729bbd1e652d8b9c18c9db28f3bc2f8a8a /tests/CaffeYolo-Armnn | |
parent | ed7fce413410d15c501ea52f9e6bfbbf71b3daf1 (diff) | |
download | armnn-6dd178f2395b34cfb360eabb0130c19ed258f5fa.tar.gz |
IVGCVSW-5720 Remove the Caffe Parser from ArmNN
Signed-off-by: Nikhil Raj <nikhil.raj@arm.com>
Change-Id: Ib00be204f549efa9aa5971ecf65c2dec4a10b10f
Diffstat (limited to 'tests/CaffeYolo-Armnn')
-rw-r--r-- | tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp | 58 |
1 files changed, 0 insertions, 58 deletions
diff --git a/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp b/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp deleted file mode 100644 index d563faaab2..0000000000 --- a/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp +++ /dev/null @@ -1,58 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// -#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>; - - int retVal = EXIT_FAILURE; - try - { - // Coverity fix: InferenceTestMain() may throw uncaught exceptions. - retVal = InferenceTestMain(argc, argv, { 0 }, - [&inputTensorShape]() - { - return make_unique<YoloTestCaseProvider<YoloInferenceModel>>( - [&] - (const InferenceTestOptions &commonOptions, - 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_InputBindings = { "data" }; - modelParams.m_OutputBindings = { "fc12" }; - modelParams.m_InputShapes = { inputTensorShape }; - modelParams.m_IsModelBinary = true; - modelParams.m_ComputeDevices = modelOptions.GetComputeDevicesAsBackendIds(); - modelParams.m_VisualizePostOptimizationModel = modelOptions.m_VisualizePostOptimizationModel; - modelParams.m_EnableFp16TurboMode = modelOptions.m_EnableFp16TurboMode; - - return std::make_unique<YoloInferenceModel>(modelParams, - commonOptions.m_EnableProfiling, - commonOptions.m_DynamicBackendsPath); - }); - }); - } - catch (const std::exception& e) - { - // Coverity fix: BOOST_LOG_TRIVIAL (typically used to report errors) may throw an - // exception of type std::length_error. - // Using stderr instead in this context as there is no point in nesting try-catch blocks here. - std::cerr << "WARNING: CaffeYolo-Armnn: An error has occurred when running " - "the classifier inference tests: " << e.what() << std::endl; - } - return retVal; -} |