Compute Library
CLHOGDescriptor Class Reference

Basic function to calculate HOG descriptor. More...

#include <CLHOGDescriptor.h>

Collaboration diagram for CLHOGDescriptor:

Public Member Functions

 CLHOGDescriptor (std::shared_ptr< IMemoryManager > memory_manager=nullptr)
 Default constructor. More...
void configure (ICLTensor *input, ICLTensor *output, const IHOG *hog, BorderMode border_mode, uint8_t constant_border_value=0)
 Initialise the function's source, destination, HOG data-object and border mode. More...
void run () override
 Run the kernels contained in the function. More...
- Public Member Functions inherited from IFunction
virtual ~IFunction ()=default
 Destructor. More...
virtual void prepare ()
 Prepare the function for executing. More...

Detailed Description

Basic function to calculate HOG descriptor.

This function calls the following OpenCL kernels:

  1. CLHOGGradient
  2. CLHOGOrientationBinningKernel
  3. CLHOGBlockNormalizationKernel

Definition at line 47 of file CLHOGDescriptor.h.

Constructor & Destructor Documentation

◆ CLHOGDescriptor()

CLHOGDescriptor ( std::shared_ptr< IMemoryManager memory_manager = nullptr)

Default constructor.

Definition at line 34 of file CLHOGDescriptor.cpp.

35  : _memory_group(std::move(memory_manager)), _gradient(), _orient_bin(), _block_norm(), _mag(), _phase(), _hog_space()
36 {
37 }

Member Function Documentation

◆ configure()

void configure ( ICLTensor input,
ICLTensor output,
const IHOG hog,
BorderMode  border_mode,
uint8_t  constant_border_value = 0 

Initialise the function's source, destination, HOG data-object and border mode.

[in,out]inputInput tensor. Data type supported: U8 (Written to only for border_mode != UNDEFINED)
[out]outputOutput tensor which stores the HOG descriptor. DataType supported: F32. The number of channels is equal to the number of histogram bins per block
[in]hogHOG data object which describes the HOG descriptor
[in]border_modeBorder mode to use.
[in]constant_border_value(Optional) Constant value to use for borders if border_mode is set to CONSTANT.

Definition at line 39 of file CLHOGDescriptor.cpp.

40 {
42  ARM_COMPUTE_ERROR_ON(nullptr == output);
43  ARM_COMPUTE_ERROR_ON(nullptr == hog);
45  const HOGInfo *hog_info = hog->info();
46  const size_t width = input->info()->dimension(Window::DimX);
47  const size_t height = input->info()->dimension(Window::DimY);
48  const size_t num_bins = hog_info->num_bins();
50  Size2D cell_size = hog_info->cell_size();
52  // Calculate number of cells along the x and y directions for the hog_space
53  const size_t num_cells_x = width / cell_size.width;
54  const size_t num_cells_y = height / cell_size.height;
56  // TensorShape of the input image
57  const TensorShape &shape_img = input->info()->tensor_shape();
59  // TensorShape of the hog space
60  TensorShape shape_hog_space = input->info()->tensor_shape();
61  shape_hog_space.set(Window::DimX, num_cells_x);
62  shape_hog_space.set(Window::DimY, num_cells_y);
64  // Intitialize tensors for magnitude, phase and hog space
65  TensorInfo info_mag(shape_img, Format::S16);
66  _mag.allocator()->init(info_mag);
68  TensorInfo info_phase(shape_img, Format::U8);
69  _phase.allocator()->init(info_phase);
71  TensorInfo info_space(shape_hog_space, num_bins, DataType::F32);
72  _hog_space.allocator()->init(info_space);
74  // Manage intermediate buffers
75  _memory_group.manage(&_mag);
76  _memory_group.manage(&_phase);
78  // Initialise gradient kernel
79  _gradient.configure(input, &_mag, &_phase, hog_info->phase_type(), border_mode, constant_border_value);
81  // Manage intermediate buffers
82  _memory_group.manage(&_hog_space);
84  // Initialise orientation binning kernel
85  _orient_bin.configure(&_mag, &_phase, &_hog_space, hog->info());
87  // Initialize HOG norm kernel
88  _block_norm.configure(&_hog_space, output, hog->info());
90  // Allocate intermediate tensors
91  _mag.allocator()->allocate();
92  _phase.allocator()->allocate();
93  _hog_space.allocator()->allocate();
94 }
size_t num_bins() const
The number of histogram bins for each cell.
Definition: HOGInfo.cpp:111
Shape of a tensor.
Definition: TensorShape.h:39
const Size2D & cell_size() const
The cell size in pixels.
Definition: HOGInfo.cpp:91
void configure(ICLTensor *input, ICLTensor *output_magnitude, ICLTensor *output_phase, PhaseType phase_type, BorderMode border_mode, uint8_t constant_border_value=0)
Initialise the function's source, destinations, phase type and border mode.
1 channel, 1 U8 per channel
Store the HOG's metadata.
Definition: HOGInfo.h:35
1 channel, 1 F32 per channel
#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
CLTensorAllocator * allocator()
Return a pointer to the tensor's allocator.
Definition: CLTensor.cpp:61
PhaseType phase_type() const
The type of PhaseType.
Definition: HOGInfo.cpp:126
void init(const TensorInfo &input, size_t alignment=0)
Initialize a tensor based on the passed TensorInfo.
size_t height
Height of the image region or rectangle.
Definition: Size2D.h:93
void manage(IMemoryManageable *obj) override
Sets a object to be managed by the given memory group.
Definition: MemoryGroup.h:79
static constexpr size_t DimX
Alias for dimension 0 also known as X dimension.
Definition: Window.h:43
void configure(const ICLTensor *input_magnitude, const ICLTensor *input_phase, ICLTensor *output, const HOGInfo *hog_info)
Initialise the kernel's inputs, output and HOG's metadata.
1 channel, 1 S16 per channel
Definition: Validate.h:790
void allocate() override
Allocate size specified by TensorInfo of OpenCL memory.
static constexpr size_t DimY
Alias for dimension 1 also known as Y dimension.
Definition: Window.h:45
size_t width
Width of the image region or rectangle.
Definition: Size2D.h:92
Class for specifying the size of an image or rectangle.
Definition: Size2D.h:34
TensorShape & set(size_t dimension, size_t value, bool apply_dim_correction=true)
Accessor to set the value of one of the dimensions.
Definition: TensorShape.h:78
void configure(const ICLTensor *input, ICLTensor *output, const HOGInfo *hog_info)
Initialise the kernel's input, output and HOG's metadata.
Store the tensor's metadata.
Definition: TensorInfo.h:45
virtual const HOGInfo * info() const =0
Interface to be implemented by the child class to return the HOG's metadata.

References CLTensorAllocator::allocate(), CLTensor::allocator(), ARM_COMPUTE_ERROR_ON, ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN, arm_compute::test::validation::border_mode, HOGInfo::cell_size(), CLHOGOrientationBinningKernel::configure(), CLHOGGradient::configure(), CLHOGBlockNormalizationKernel::configure(), Window::DimX, Window::DimY, arm_compute::F32, Size2D::height, IHOG::info(), ITensorAllocator::init(), arm_compute::test::validation::input, MemoryGroup::manage(), HOGInfo::num_bins(), HOGInfo::phase_type(), arm_compute::S16, TensorShape::set(), arm_compute::U8, and Size2D::width.

◆ run()

void run ( )

Run the kernels contained in the function.

For NEON kernels:

  • Multi-threading is used for the kernels which are parallelisable.
  • By default std::thread::hardware_concurrency() threads are used.
CPPScheduler::set_num_threads() can be used to manually set the number of threads

For OpenCL kernels:

  • All the kernels are enqueued on the queue associated with CLScheduler.
  • The queue is then flushed.
The function will not block until the kernels are executed. It is the user's responsibility to wait.
Will call prepare() on first run if hasn't been done

Implements IFunction.

Definition at line 96 of file CLHOGDescriptor.cpp.

97 {
98  MemoryGroupResourceScope scope_mg(_memory_group);
100  // Run gradient
103  // Run orientation binning
104  CLScheduler::get().enqueue(_orient_bin, false);
106  // Run block normalization
107  CLScheduler::get().enqueue(_block_norm);
108 }
static CLScheduler & get()
Access the scheduler singleton.
Definition: CLScheduler.cpp:99
void enqueue(ICLKernel &kernel, bool flush=true)
Schedule the execution of the passed kernel if possible.
Memory group resources scope handling class.
Definition: IMemoryGroup.h:82
void run() override
Run the kernels contained in the function.

References CLScheduler::enqueue(), CLScheduler::get(), and CLHOGGradient::run().

The documentation for this class was generated from the following files: