diff options
author | Pablo Tello <pablo.tello@arm.com> | 2018-11-15 14:43:10 +0000 |
---|---|---|
committer | Anthony Barbier <Anthony.barbier@arm.com> | 2018-11-19 09:58:20 +0000 |
commit | 32521430ad67172d372721979e8a32b7d4317620 (patch) | |
tree | c18921290341c1d730fbff406960dc7ea7f61306 /utils/GraphUtils.h | |
parent | d696cb6d18c2fe66f1abce88bbd14faf2137ef89 (diff) | |
download | ComputeLibrary-32521430ad67172d372721979e8a32b7d4317620.tar.gz |
COMPMID-1750: Add PriorBox operator to graph API.
Change-Id: I5d2ed5dcc342abff8124762f7bdee587cdf20032
Diffstat (limited to 'utils/GraphUtils.h')
-rw-r--r-- | utils/GraphUtils.h | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/utils/GraphUtils.h b/utils/GraphUtils.h index a6d670d761..d7f24afdd8 100644 --- a/utils/GraphUtils.h +++ b/utils/GraphUtils.h @@ -63,14 +63,16 @@ 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 */ - CaffePreproccessor(std::array<float, 3> mean = std::array<float, 3> { { 0, 0, 0 } }, bool bgr = true); + CaffePreproccessor(std::array<float, 3> mean = std::array<float, 3> { { 0, 0, 0 } }, float scale = 1.f, bool bgr = true); void preprocess(ITensor &tensor) override; private: std::array<float, 3> _mean; - bool _bgr; + float _scale; + bool _bgr; }; /** TF preproccessor */ |