2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // See LICENSE file in the project root for full license information.
7 #include "ClassifierTestCaseData.hpp"
14 using ImageSet = std::pair<const std::string, unsigned int>;
16 class MobileNetDatabase
19 using TTestCaseData = ClassifierTestCaseData<float>;
21 explicit MobileNetDatabase(const std::string& binaryFileDirectory,
24 const std::vector<ImageSet>& imageSet);
26 std::unique_ptr<TTestCaseData> GetTestCaseData(unsigned int testCaseId);
29 unsigned int GetNumImageElements() const { return 3 * m_Width * m_Height; }
30 unsigned int GetNumImageBytes() const { return 4 * GetNumImageElements(); }
32 std::string m_BinaryDirectory;
33 unsigned int m_Height;
35 const std::vector<ImageSet> m_ImageSet;