forked from Qortal/Brooklyn
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24 lines
537 B
24 lines
537 B
// |
|
// Copyright © 2017 Arm Ltd. All rights reserved. |
|
// SPDX-License-Identifier: MIT |
|
// |
|
#pragma once |
|
|
|
#include "ClassifierTestCaseData.hpp" |
|
|
|
#include <string> |
|
#include <memory> |
|
|
|
class Cifar10Database |
|
{ |
|
public: |
|
using DataType = float; |
|
using TTestCaseData = ClassifierTestCaseData<DataType>; |
|
|
|
explicit Cifar10Database(const std::string& binaryFileDirectory, bool rgbPack = false); |
|
std::unique_ptr<TTestCaseData> GetTestCaseData(unsigned int testCaseId); |
|
|
|
private: |
|
std::string m_BinaryDirectory; |
|
bool m_RgbPack; |
|
};
|
|
|