wip #63

Author: previ

camera.py

@@ -74,7 +74,7 @@ def __init__(self):
         if (PHOTO_PREFIX in filename or VIDEO_PREFIX in filename) and PHOTO_THUMB_SUFFIX not in filename:
           self._photos.append(filename)
 
-    self._cnn_classifier = CNNClassifier("cnn_models/applekiwi.pb", "cnn_models/applekiwi.txt", "input", "final_result", 128, 128, 0.0, 255.0)
+    self._cnn_classifier = CNNClassifier("cnn_models/applekiwi_1_128.pb", "cnn_models/applekiwi.txt", "input", "final_result", 128, 128, 0.0, 255.0)
 
     super(Camera, self).__init__()
 

cnn_classifier.py

@@ -24,7 +24,7 @@
 import tensorflow as tf
 
 class CNNClassifier:
-    def __init__(self, model_file, label_file, input_layer="input", output_layer="final_output", input_height=128, input_width=128, input_mean=0.0, input_std=255.0):
+    def __init__(self, model_file, label_file, input_layer="input", output_layer="final_output", input_height=128, input_width=128, input_mean=127.5, input_std=127.5):
         self._graph = self.load_graph(model_file)
         self._labels = self.load_labels(label_file)
         input_name = "import/" + input_layer