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/Cifar10Database.cpp | |
download | armnn-4fcda0101ec3d110c1d6d7bee5c83416b645528a.tar.gz |
Release 18.02
Change-Id: Id3c11dc5ee94ef664374a988fcc6901e9a232fa6
Diffstat (limited to 'tests/Cifar10Database.cpp')
-rw-r--r-- | tests/Cifar10Database.cpp | 84 |
1 files changed, 84 insertions, 0 deletions
diff --git a/tests/Cifar10Database.cpp b/tests/Cifar10Database.cpp new file mode 100644 index 0000000000..f3bf68fd45 --- /dev/null +++ b/tests/Cifar10Database.cpp @@ -0,0 +1,84 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// See LICENSE file in the project root for full license information. +// +#include "Cifar10Database.hpp" + +#include <boost/numeric/conversion/cast.hpp> +#include <boost/log/trivial.hpp> +#include <fstream> +#include <vector> + +constexpr unsigned int g_kCifar10ImageByteSize = 1 + 3 * 32 * 32; + +Cifar10Database::Cifar10Database(const std::string& binaryFileDirectory, bool rgbPack) + : m_BinaryDirectory(binaryFileDirectory), m_RgbPack(rgbPack) +{ +} + +std::unique_ptr<Cifar10Database::TTestCaseData> Cifar10Database::GetTestCaseData(unsigned int testCaseId) +{ + std::vector<unsigned char> I(g_kCifar10ImageByteSize); + + std::string fullpath = m_BinaryDirectory + std::string("test_batch.bin"); + + std::ifstream fileStream(fullpath, std::ios::binary); + if (!fileStream.is_open()) + { + BOOST_LOG_TRIVIAL(fatal) << "Failed to load " << fullpath; + return nullptr; + } + + fileStream.seekg(testCaseId * g_kCifar10ImageByteSize, std::ios_base::beg); + fileStream.read(reinterpret_cast<char*>(&I[0]), g_kCifar10ImageByteSize); + + if (!fileStream.good()) + { + BOOST_LOG_TRIVIAL(fatal) << "Failed to read " << fullpath; + return nullptr; + } + + + std::vector<float> inputImageData; + inputImageData.resize(g_kCifar10ImageByteSize - 1); + + unsigned int step; + unsigned int countR_o; + unsigned int countG_o; + unsigned int countB_o; + unsigned int countR = 1; + unsigned int countG = 1 + 32 * 32; + unsigned int countB = 1 + 2 * 32 * 32; + + if (m_RgbPack) + { + countR_o = 0; + countG_o = 1; + countB_o = 2; + step = 3; + } + else + { + countR_o = 0; + countG_o = 32 * 32; + countB_o = 2 * 32 * 32; + step = 1; + } + + for (unsigned int h = 0; h < 32; h++) + { + for (unsigned int w = 0; w < 32; w++) + { + inputImageData[countR_o] = boost::numeric_cast<float>(I[countR++]); + inputImageData[countG_o] = boost::numeric_cast<float>(I[countG++]); + inputImageData[countB_o] = boost::numeric_cast<float>(I[countB++]); + + countR_o += step; + countG_o += step; + countB_o += step; + } + } + + const unsigned int label = boost::numeric_cast<unsigned int>(I[0]); + return std::make_unique<TTestCaseData>(label, std::move(inputImageData)); +} |