Fix normalization error from TF Lite codelab as models from Model Mak…

…er use [0, 1] normalization. PiperOrigin-RevId: 319352280
hupnos · Jul 2, 2020 · bb26df4 · bb26df4
1 parent 9d70a17
commit bb26df4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/...ain/java/org/tensorflow/lite/examples/classification/tflite/ClassifierFloatMobileNet.java b/...ain/java/org/tensorflow/lite/examples/classification/tflite/ClassifierFloatMobileNet.java
@@ -25,9 +25,9 @@
 public class ClassifierFloatMobileNet extends Classifier {
 
   /** Float MobileNet requires additional normalization of the used input. */
-  private static final float IMAGE_MEAN = 127.5f;
+  private static final float IMAGE_MEAN = 0f;
 
-  private static final float IMAGE_STD = 127.5f;
+  private static final float IMAGE_STD = 255f;
 
   /**
    * Float model does not need dequantization in the post-processing. Setting mean and std as 0.0f

diff --git a/...ain/java/org/tensorflow/lite/examples/classification/tflite/ClassifierFloatMobileNet.java b/...ain/java/org/tensorflow/lite/examples/classification/tflite/ClassifierFloatMobileNet.java
@@ -25,9 +25,9 @@
 public class ClassifierFloatMobileNet extends Classifier {
 
   /** Float MobileNet requires additional normalization of the used input. */
-  private static final float IMAGE_MEAN = 127.5f;
+  private static final float IMAGE_MEAN = 0f;
 
-  private static final float IMAGE_STD = 127.5f;
+  private static final float IMAGE_STD = 255f;
 
   /**
    * Float model does not need dequantization in the post-processing. Setting mean and std as 0.0f