Compute Library
18.05
|
Caffe preproccessor. More...
#include <GraphUtils.h>
Public Member Functions | |
CaffePreproccessor (std::array< float, 3 > mean=std::array< float, 3 >{{0, 0, 0}}, bool bgr=true) | |
Default Constructor. More... | |
void | preprocess (ITensor &tensor) override |
Preprocess the given tensor. More... | |
Public Member Functions inherited from IPreprocessor | |
virtual | ~IPreprocessor ()=default |
Default destructor. More... | |
Caffe preproccessor.
Definition at line 57 of file GraphUtils.h.
CaffePreproccessor | ( | std::array< float, 3 > | mean = std::array<float, 3> { { 0, 0, 0 } } , |
bool | bgr = true |
||
) |
Default Constructor.
mean | Mean array in RGB ordering |
bgr | Boolean specifying if the preprocessing should assume BGR format |
Definition at line 71 of file GraphUtils.cpp.
References arm_compute::test::swap().
|
overridevirtual |
Preprocess the given tensor.
[in] | tensor | Tensor to preprocess. |
Implements IPreprocessor.
Definition at line 80 of file GraphUtils.cpp.
References arm_compute::execute_window_loop(), ITensor::info(), ITensor::ptr_to_element(), ITensorInfo::tensor_shape(), and Window::use_tensor_dimensions().