2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
7 #include "ClassifierTestCaseData.hpp"
13 struct YoloBoundingBox
21 struct YoloDetectedObject
23 YoloDetectedObject(unsigned int yoloClass,
24 const YoloBoundingBox& box,
28 , m_Confidence(confidence)
32 YoloBoundingBox m_Box;
36 class YoloTestCaseData
39 YoloTestCaseData(std::vector<float> inputImage,
40 std::vector<YoloDetectedObject> topObjectDetections)
41 : m_InputImage(std::move(inputImage))
42 , m_TopObjectDetections(std::move(topObjectDetections))
46 std::vector<float> m_InputImage;
47 std::vector<YoloDetectedObject> m_TopObjectDetections;
50 constexpr unsigned int YoloImageWidth = 448;
51 constexpr unsigned int YoloImageHeight = 448;
56 using TTestCaseData = YoloTestCaseData;
58 explicit YoloDatabase(const std::string& imageDir);
59 std::unique_ptr<TTestCaseData> GetTestCaseData(unsigned int testCaseId);
62 std::string m_ImageDir;