1 // 2 // Copyright © 2017 Arm Ltd. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 #pragma once 6 7 #include "ClassifierTestCaseData.hpp" 8 9 #include <string> 10 #include <memory> 11 12 class MnistDatabase 13 { 14 public: 15 using DataType = float; 16 using TTestCaseData = ClassifierTestCaseData<DataType>; 17 18 explicit MnistDatabase(const std::string& binaryFileDirectory, bool scaleValues = false); 19 std::unique_ptr<TTestCaseData> GetTestCaseData(unsigned int testCaseId); 20 21 private: 22 std::string m_BinaryDirectory; 23 bool m_ScaleValues; 24 };