diff options
Diffstat (limited to 'utils/GraphUtils.h')
| -rw-r--r-- | utils/GraphUtils.h | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/utils/GraphUtils.h b/utils/GraphUtils.h index 4ae484f430..88221c7dc8 100644 --- a/utils/GraphUtils.h +++ b/utils/GraphUtils.h @@ -62,17 +62,17 @@ class CaffePreproccessor : public IPreprocessor public: /** Default Constructor * - * @param mean Mean array in RGB ordering - * @param scale Scale value - * @param bgr Boolean specifying if the preprocessing should assume BGR format + * @param[in] mean Mean array in RGB ordering + * @param[in] bgr Boolean specifying if the preprocessing should assume BGR format + * @param[in] scale Scale value */ - CaffePreproccessor(std::array<float, 3> mean = std::array<float, 3> { { 0, 0, 0 } }, float scale = 1.f, bool bgr = true); + CaffePreproccessor(std::array<float, 3> mean = std::array<float, 3> { { 0, 0, 0 } }, bool bgr = true, float scale = 1.f); void preprocess(ITensor &tensor) override; private: std::array<float, 3> _mean; - float _scale; bool _bgr; + float _scale; }; /** TF preproccessor */ |