diff options
author | telsoa01 <telmo.soares@arm.com> | 2018-03-09 14:13:49 +0000 |
---|---|---|
committer | telsoa01 <telmo.soares@arm.com> | 2018-03-09 14:13:49 +0000 |
commit | 4fcda0101ec3d110c1d6d7bee5c83416b645528a (patch) | |
tree | c9a70aeb2887006160c1b3d265c27efadb7bdbae /tests/ImageNetDatabase.cpp | |
download | armnn-4fcda0101ec3d110c1d6d7bee5c83416b645528a.tar.gz |
Release 18.02
Change-Id: Id3c11dc5ee94ef664374a988fcc6901e9a232fa6
Diffstat (limited to 'tests/ImageNetDatabase.cpp')
-rw-r--r-- | tests/ImageNetDatabase.cpp | 54 |
1 files changed, 54 insertions, 0 deletions
diff --git a/tests/ImageNetDatabase.cpp b/tests/ImageNetDatabase.cpp new file mode 100644 index 0000000000..0a235c9a3e --- /dev/null +++ b/tests/ImageNetDatabase.cpp @@ -0,0 +1,54 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// See LICENSE file in the project root for full license information. +// +#include "InferenceTestImage.hpp" +#include "ImageNetDatabase.hpp" + +#include <boost/numeric/conversion/cast.hpp> +#include <boost/log/trivial.hpp> +#include <boost/assert.hpp> +#include <boost/format.hpp> + +#include <iostream> +#include <fcntl.h> +#include <array> + +const std::vector<ImageSet> g_DefaultImageSet = +{ + {"shark.jpg", 2} +}; + +ImageNetDatabase::ImageNetDatabase(const std::string& binaryFileDirectory, unsigned int width, unsigned int height, + const std::vector<ImageSet>& imageSet) +: m_BinaryDirectory(binaryFileDirectory) +, m_Height(height) +, m_Width(width) +, m_ImageSet(imageSet.empty() ? g_DefaultImageSet : imageSet) +{ +} + +std::unique_ptr<ImageNetDatabase::TTestCaseData> ImageNetDatabase::GetTestCaseData(unsigned int testCaseId) +{ + testCaseId = testCaseId % boost::numeric_cast<unsigned int>(m_ImageSet.size()); + const ImageSet& imageSet = m_ImageSet[testCaseId]; + const std::string fullPath = m_BinaryDirectory + imageSet.first; + FILE* file = fopen(fullPath.c_str(), "rb"); + + if (file == nullptr) + { + BOOST_LOG_TRIVIAL(fatal) << "Failed to load " << fullPath; + return nullptr; + } + + InferenceTestImage image(fullPath.c_str()); + image.Resize(m_Width, m_Height); + + // The model expects image data in BGR format + std::vector<float> 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<TTestCaseData>(label, std::move(inputImageData)); +} |