| // |
| // Copyright © 2017 Arm Ltd. All rights reserved. |
| // See LICENSE file in the project root for full license information. |
| // |
| #pragma once |
| |
| #include "ClassifierTestCaseData.hpp" |
| |
| #include <string> |
| #include <memory> |
| |
| class MnistDatabase |
| { |
| public: |
| using TTestCaseData = ClassifierTestCaseData<float>; |
| |
| explicit MnistDatabase(const std::string& binaryFileDirectory, bool scaleValues = false); |
| std::unique_ptr<TTestCaseData> GetTestCaseData(unsigned int testCaseId); |
| |
| private: |
| std::string m_BinaryDirectory; |
| bool m_ScaleValues; |
| }; |