diff options
Diffstat (limited to 'utils/GraphUtils.h')
-rw-r--r-- | utils/GraphUtils.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/utils/GraphUtils.h b/utils/GraphUtils.h index 3417135f17..4c25dd2460 100644 --- a/utils/GraphUtils.h +++ b/utils/GraphUtils.h @@ -70,6 +70,9 @@ public: void preprocess(ITensor &tensor) override; private: + template <typename T> + void preprocess_typed(ITensor &tensor); + std::array<float, 3> _mean; bool _bgr; float _scale; @@ -90,6 +93,9 @@ public: void preprocess(ITensor &tensor) override; private: + template <typename T> + void preprocess_typed(ITensor &tensor); + float _min_range; float _max_range; }; |