tests/YoloDatabase.cpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // See LICENSE file in the project root for full license information.
   4 //
   5 #include "YoloDatabase.hpp"
   6
   7 #include <armnn/Exceptions.hpp>
   8
   9 #include <array>
  10 #include <cstdint>
  11 #include <tuple>
  12 #include <utility>
  13
  14 #include <boost/assert.hpp>
  15 #include <boost/format.hpp>
  16 #include <boost/log/trivial.hpp>
  17 #include <boost/numeric/conversion/cast.hpp>
  18
  19 #include "InferenceTestImage.hpp"
  20
  21 namespace
  22 {
  23 enum class YoloVocClass : unsigned int
  24 {
  25     Aeroplane,
  26     Bicycle,
  27     Bird,
  28     Boat,
  29     Bottle,
  30     Bus,
  31     Car,
  32     Cat,
  33     Chair,
  34     Cow,
  35     DiningTable,
  36     Dog,
  37     Horse,
  38     Motorbike,
  39     Person,
  40     PottedPlant,
  41     Sheep,
  42     Sofa,
  43     Train,
  44     TvMonitor
  45 };
  46
  47 template <typename E>
  48 constexpr auto to_underlying(E e) noexcept
  49 {
  50     return static_cast<std::underlying_type_t<E>>(e);
  51 }
  52
  53 class ImageNotFoundException : public armnn::Exception
  54 {
  55     using Exception::Exception;
  56 };
  57
  58 using YoloInputOutput = std::pair<const char* const, YoloDetectedObject>;
  59
  60 const std::array<YoloInputOutput,1> g_PerTestCaseInputOutput =
  61 {
  62     YoloInputOutput{
  63         "yolo_dog_448x448.png",
  64         { to_underlying(YoloVocClass::Dog), YoloBoundingBox{ 233.0f, 256.0f, 299.0f, 462.0f }, 0.5088733434677124f }
  65     },
  66 };
  67
  68 } // namespace
  69
  70 YoloDatabase::YoloDatabase(const std::string& imageDir)
  71     : m_ImageDir(imageDir)
  72 {
  73 }
  74
  75 std::unique_ptr<YoloDatabase::TTestCaseData> YoloDatabase::GetTestCaseData(unsigned int testCaseId)
  76 {
  77     testCaseId = testCaseId % boost::numeric_cast<unsigned int>(g_PerTestCaseInputOutput.size());
  78     const auto& testCaseInputOutput = g_PerTestCaseInputOutput[testCaseId];
  79     const std::string imagePath = m_ImageDir + testCaseInputOutput.first;
  80
  81     // Load test case input image
  82     std::vector<float> imageData;
  83     try
  84     {
  85         InferenceTestImage image(imagePath.c_str());
  86         image.Resize(YoloImageWidth, YoloImageHeight);
  87         imageData = GetImageDataInArmNnLayoutAsNormalizedFloats(ImageChannelLayout::Rgb, image);
  88     }
  89     catch (const InferenceTestImageException& e)
  90     {
  91         BOOST_LOG_TRIVIAL(fatal) << "Failed to load test case " << testCaseId << " with error: " << e.what();
  92         return nullptr;
  93     }
  94
  95     // Prepare test case output
  96     std::vector<YoloDetectedObject> topObjectDetections;
  97     topObjectDetections.reserve(1);
  98     topObjectDetections.push_back(testCaseInputOutput.second);
  99
 100     return std::make_unique<YoloTestCaseData>(std::move(imageData), std::move(topObjectDetections));
 101 }