tests/Cifar10Database.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 Cifar10Database
  13 {
  14 public:
  15     using DataType = float;
  16     using TTestCaseData = ClassifierTestCaseData<DataType>;
  17
  18     explicit Cifar10Database(const std::string& binaryFileDirectory, bool rgbPack = false);
  19     std::unique_ptr<TTestCaseData> GetTestCaseData(unsigned int testCaseId);
  20
  21 private:
  22     std::string m_BinaryDirectory;
  23     bool m_RgbPack;
  24 };