From 6dd178f2395b34cfb360eabb0130c19ed258f5fa Mon Sep 17 00:00:00 2001 From: Nikhil Raj Date: Fri, 2 Apr 2021 22:04:39 +0100 Subject: IVGCVSW-5720 Remove the Caffe Parser from ArmNN Signed-off-by: Nikhil Raj Change-Id: Ib00be204f549efa9aa5971ecf65c2dec4a10b10f --- tests/CaffePreprocessor.cpp | 44 -------------------------------------------- 1 file changed, 44 deletions(-) delete mode 100644 tests/CaffePreprocessor.cpp (limited to 'tests/CaffePreprocessor.cpp') diff --git a/tests/CaffePreprocessor.cpp b/tests/CaffePreprocessor.cpp deleted file mode 100644 index 54ce833b72..0000000000 --- a/tests/CaffePreprocessor.cpp +++ /dev/null @@ -1,44 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// -#include "InferenceTestImage.hpp" -#include "CaffePreprocessor.hpp" - -#include - -#include -#include -#include - -const std::vector g_DefaultImageSet = -{ - {"shark.jpg", 2} -}; - -CaffePreprocessor::CaffePreprocessor(const std::string& binaryFileDirectory, unsigned int width, unsigned int height, - const std::vector& imageSet) -: m_BinaryDirectory(binaryFileDirectory) -, m_Height(height) -, m_Width(width) -, m_ImageSet(imageSet.empty() ? g_DefaultImageSet : imageSet) -{ -} - -std::unique_ptr CaffePreprocessor::GetTestCaseData(unsigned int testCaseId) -{ - testCaseId = testCaseId % armnn::numeric_cast(m_ImageSet.size()); - const ImageSet& imageSet = m_ImageSet[testCaseId]; - const std::string fullPath = m_BinaryDirectory + imageSet.first; - - InferenceTestImage image(fullPath.c_str()); - image.Resize(m_Width, m_Height, CHECK_LOCATION()); - - // The model expects image data in BGR format. - std::vector inputImageData = GetImageDataInArmNnLayoutAsFloatsSubtractingMean(ImageChannelLayout::Bgr, - image, m_MeanBgr); - - // List of labels: https://gist.github.com/yrevar/942d3a0ac09ec9e5eb3a - const unsigned int label = imageSet.second; - return std::make_unique(label, std::move(inputImageData)); -} -- cgit v1.2.1