tests/MnistDatabase.hpp

   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 };