ArmNN  NotReleased
ImagePreprocessor.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 #pragma once
6 
8 
9 #include <array>
10 #include <string>
11 #include <vector>
12 #include <memory>
13 
15 
16 
17 using ImageSet = std::pair<const std::string, unsigned int>;
18 
19 template <typename TDataType>
21 {
22 public:
23  using DataType = TDataType;
25 
27  {
30  };
31 
32  explicit ImagePreprocessor(const std::string& binaryFileDirectory,
33  unsigned int width,
34  unsigned int height,
35  const std::vector<ImageSet>& imageSet,
36  float scale=255.0f,
37  const std::array<float, 3> mean={{0, 0, 0}},
38  const std::array<float, 3> stddev={{1, 1, 1}},
39  DataFormat dataFormat=DataFormat::NHWC,
40  unsigned int batchSize=1)
41  : m_BinaryDirectory(binaryFileDirectory)
42  , m_Height(height)
43  , m_Width(width)
44  , m_BatchSize(batchSize)
45  , m_Scale(scale)
46  , m_ImageSet(imageSet)
47  , m_Mean(mean)
48  , m_Stddev(stddev)
49  , m_DataFormat(dataFormat)
50  {
51  }
52 
53  std::unique_ptr<TTestCaseData> GetTestCaseData(unsigned int testCaseId);
54 
55 private:
56  unsigned int GetNumImageElements() const { return 3 * m_Width * m_Height; }
57  unsigned int GetNumImageBytes() const { return sizeof(DataType) * GetNumImageElements(); }
58  unsigned int GetLabelAndResizedImageAsFloat(unsigned int testCaseId,
59  std::vector<float> & result);
60 
61  std::string m_BinaryDirectory;
62  unsigned int m_Height;
63  unsigned int m_Width;
64  unsigned int m_BatchSize;
65  // Quantization parameters
66  float m_Scale;
67  const std::vector<ImageSet> m_ImageSet;
68 
69  const std::array<float, 3> m_Mean;
70  const std::array<float, 3> m_Stddev;
71 
72  DataFormat m_DataFormat;
73 };
ImagePreprocessor(const std::string &binaryFileDirectory, unsigned int width, unsigned int height, const std::vector< ImageSet > &imageSet, float scale=255.0f, const std::array< float, 3 > mean={{0, 0, 0}}, const std::array< float, 3 > stddev={{1, 1, 1}}, DataFormat dataFormat=DataFormat::NHWC, unsigned int batchSize=1)
std::unique_ptr< TTestCaseData > GetTestCaseData(unsigned int testCaseId)
std::pair< const std::string, unsigned int > ImageSet
Caffe requires BGR images, not normalized, mean adjusted and resized using smooth resize of STB libra...