Compute Library
 21.02
Pyramid.cpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2016-2020 Arm Limited.
3  *
4  * SPDX-License-Identifier: MIT
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy
7  * of this software and associated documentation files (the "Software"), to
8  * deal in the Software without restriction, including without limitation the
9  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10  * sell copies of the Software, and to permit persons to whom the Software is
11  * furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included in all
14  * copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22  * SOFTWARE.
23  */
25 
26 #include "arm_compute/core/Error.h"
30 
31 #include <cmath>
32 
33 using namespace arm_compute;
34 
36 {
37  internal_init(info, false);
38 }
39 
41 {
42  internal_init(info, true);
43 }
44 
45 void Pyramid::internal_init(const PyramidInfo &info, bool auto_padding)
46 {
47  _info = info;
48  _pyramid.resize(_info.num_levels());
49 
50  size_t w = _info.width();
51  size_t h = _info.height();
52  size_t ref_w = w;
53  size_t ref_h = h;
54  bool is_orb_scale = (SCALE_PYRAMID_ORB == _info.scale());
55  TensorShape tensor_shape = _info.tensor_shape();
56 
57  // Note: Look-up table used by the OpenVX sample implementation
58  const std::array<float, 4> c_orbscale = { 0.5f,
61  SCALE_PYRAMID_ORB *SCALE_PYRAMID_ORB * SCALE_PYRAMID_ORB
62  };
63 
64  for(size_t i = 0; i < _info.num_levels(); ++i)
65  {
66  TensorInfo tensor_info(tensor_shape, _info.format());
67 
68  if(auto_padding)
69  {
70  tensor_info.auto_padding();
71  }
72 
73  _pyramid[i].allocator()->init(tensor_info);
74 
75  if(is_orb_scale)
76  {
77  float orb_scale = c_orbscale[(i + 1) % 4];
78  w = static_cast<int>(std::ceil(static_cast<float>(ref_w) * orb_scale));
79  h = static_cast<int>(std::ceil(static_cast<float>(ref_h) * orb_scale));
80 
81  if(0 == ((i + 1) % 4))
82  {
83  ref_w = w;
84  ref_h = h;
85  }
86  }
87  else
88  {
89  w = (w + 1) * _info.scale();
90  h = (h + 1) * _info.scale();
91  }
92 
93  // Update tensor_shape
94  tensor_shape.set(0, w);
95  tensor_shape.set(1, h);
96  }
97 }
98 
100 {
101  for(size_t i = 0; i < _info.num_levels(); ++i)
102  {
103  _pyramid[i].allocator()->allocate();
104  }
105 }
106 
108 {
109  return &_info;
110 }
111 
112 Tensor *Pyramid::get_pyramid_level(size_t index) const
113 {
114  ARM_COMPUTE_ERROR_ON(index >= _info.num_levels());
115 
116  return &_pyramid[index];
117 }
constexpr float SCALE_PYRAMID_ORB
Constant value used to indicate a ORB scaled pyramid.
Definition: Types.h:115
const TensorShape & tensor_shape() const
Return the TensorShape of the o-th level tensor.
Definition: PyramidInfo.cpp:92
SimpleTensor< float > w
Definition: DFT.cpp:156
Shape of a tensor.
Definition: TensorShape.h:39
Tensor * get_pyramid_level(size_t index) const override
Retrieves a level of the pyramid as a ITensor pointer.
Definition: Pyramid.cpp:112
Format format() const
Return the image format of all tensor in the pyramid.
Definition: PyramidInfo.cpp:97
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
Definition: Error.h:466
bool auto_padding() override
Update the offset to the first element and the strides to automatically computed values.
Definition: TensorInfo.cpp:224
Copyright (c) 2017-2021 Arm Limited.
void init(const PyramidInfo &info)
Initialize pyramid data-object using the given Pyramid&#39;s metadata.
Definition: Pyramid.cpp:35
void init_auto_padding(const PyramidInfo &info)
Initialize pyramid data-object using the given Pyramid&#39;s metadata.
Definition: Pyramid.cpp:40
Basic implementation of the tensor interface.
Definition: Tensor.h:37
float scale() const
Return the scale factor of the pyramid.
void allocate()
Allocate the planes in the pyramid.
Definition: Pyramid.cpp:99
const PyramidInfo * info() const override
Interface to be implemented by the child class to return the Pyramid&#39;s metadata.
Definition: Pyramid.cpp:107
Store the Pyramid&#39;s metadata.
Definition: PyramidInfo.h:35
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
Store the tensor&#39;s metadata.
Definition: TensorInfo.h:45
size_t width() const
Return the width of the 0th level tensor.
Definition: PyramidInfo.cpp:82
size_t num_levels() const
Return the number of the pyramid levels.
Definition: PyramidInfo.cpp:77
size_t height() const
Return the height of the 0th level tensor.
Definition: PyramidInfo.cpp:87
TensorShape & set(size_t dimension, size_t value, bool apply_dim_correction=true, bool increase_dim_unit=true)
Accessor to set the value of one of the dimensions.
Definition: TensorShape.h:79