diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-08-17 15:33:39 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
commit | be2772a6b0b88cc3170f6527e07e8bcb426a4b1c (patch) | |
tree | e806f0f2c52360a16bf456be4ec94b5c56c9f3fd /utils/GraphUtils.cpp | |
parent | 6c95c2dd574ebc3217c949a17016eb071935bc3b (diff) | |
download | ComputeLibrary-be2772a6b0b88cc3170f6527e07e8bcb426a4b1c.tar.gz |
COMPMID-1508: Add Inception ResNet V2 model
Change-Id: Iab860a43aa831690fab49b96c124528cc4cb14f2
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/144621
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Diffstat (limited to 'utils/GraphUtils.cpp')
-rw-r--r-- | utils/GraphUtils.cpp | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/utils/GraphUtils.cpp b/utils/GraphUtils.cpp index f07323cdf8..f33d8a2e2f 100644 --- a/utils/GraphUtils.cpp +++ b/utils/GraphUtils.cpp @@ -57,16 +57,22 @@ std::pair<arm_compute::TensorShape, arm_compute::PermutationVector> compute_perm } } // namespace +TFPreproccessor::TFPreproccessor(float min_range, float max_range) + : _min_range(min_range), _max_range(max_range) +{ +} void TFPreproccessor::preprocess(ITensor &tensor) { Window window; window.use_tensor_dimensions(tensor.info()->tensor_shape()); + const float range = _max_range - _min_range; + execute_window_loop(window, [&](const Coordinates & id) { const float value = *reinterpret_cast<float *>(tensor.ptr_to_element(id)); - float res = value / 255.f; // Normalize to [0, 1] - res = (res - 0.5f) * 2.f; // Map to [-1, 1] + float res = value / 255.f; // Normalize to [0, 1] + res = res * range + _min_range; // Map to [min_range, max_range] *reinterpret_cast<float *>(tensor.ptr_to_element(id)) = res; }); } |