From be2772a6b0b88cc3170f6527e07e8bcb426a4b1c Mon Sep 17 00:00:00 2001
From: Georgios Pinitas <georgios.pinitas@arm.com>
Date: Fri, 17 Aug 2018 15:33:39 +0100
Subject: 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>
---
 utils/GraphUtils.cpp | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

(limited to 'utils/GraphUtils.cpp')

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;
     });
 }
-- 
cgit v1.2.1