diff options
Diffstat (limited to 'tests/YoloDatabase.hpp')
-rw-r--r-- | tests/YoloDatabase.hpp | 63 |
1 files changed, 0 insertions, 63 deletions
diff --git a/tests/YoloDatabase.hpp b/tests/YoloDatabase.hpp deleted file mode 100644 index 220262c522..0000000000 --- a/tests/YoloDatabase.hpp +++ /dev/null @@ -1,63 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// -#pragma once - -#include "ClassifierTestCaseData.hpp" - -#include <array> -#include <string> -#include <memory> - -struct YoloBoundingBox -{ - float m_X; - float m_Y; - float m_W; - float m_H; -}; - -struct YoloDetectedObject -{ - YoloDetectedObject(unsigned int yoloClass, - const YoloBoundingBox& box, - float confidence) - : m_Class(yoloClass) - , m_Box(box) - , m_Confidence(confidence) - {} - - unsigned int m_Class; - YoloBoundingBox m_Box; - float m_Confidence; -}; - -class YoloTestCaseData -{ -public: - YoloTestCaseData(std::vector<float> inputImage, - std::vector<YoloDetectedObject> topObjectDetections) - : m_InputImage(std::move(inputImage)) - , m_TopObjectDetections(std::move(topObjectDetections)) - { - } - - std::vector<float> m_InputImage; - std::vector<YoloDetectedObject> m_TopObjectDetections; -}; - -constexpr unsigned int YoloImageWidth = 448; -constexpr unsigned int YoloImageHeight = 448; - -class YoloDatabase -{ -public: - using TTestCaseData = YoloTestCaseData; - - explicit YoloDatabase(const std::string& imageDir); - std::unique_ptr<TTestCaseData> GetTestCaseData(unsigned int testCaseId); - -private: - std::string m_ImageDir; -}; |