24.02.1
|
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, float scale=1.f) | |
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 60 of file GraphUtils.h.
CaffePreproccessor | ( | std::array< float, 3 > | mean = std::array<float, 3>{{0, 0, 0}} , |
bool | bgr = true , |
||
float | scale = 1.f |
||
) |
Default Constructor.
[in] | mean | Mean array in RGB ordering |
[in] | bgr | Boolean specifying if the preprocessing should assume BGR format |
[in] | scale | Scale value |
Definition at line 104 of file GraphUtils.cpp.
References arm_compute::swap().
|
overridevirtual |
Preprocess the given tensor.
[in] | tensor | Tensor to preprocess. |
Implements IPreprocessor.
Definition at line 113 of file GraphUtils.cpp.
References ARM_COMPUTE_ERROR, arm_compute::F16, arm_compute::F32, and tensor.