Merge pull request #40 from nanowireUK/network

Read placeholder size from graph

Author: emmanuel-bv

modules/ImageClassifierService/app/predict.py

@@ -13,7 +13,6 @@
 filename = 'model.pb'
 labels_filename = 'labels.txt'
 
-network_input_size = 227
 mean_values_b_g_r = (0,0,0)
 
 size = (256, 256)
@@ -58,6 +57,9 @@ def predict_image(image):
     with tf.Session() as sess:
         prob_tensor = sess.graph.get_tensor_by_name(output_layer)
 
+        input_tensor_shape = sess.graph.get_tensor_by_name('Placeholder:0').shape.as_list()
+        network_input_size = input_tensor_shape[1]
+
         # w = image.shape[0]
         # h = image.shape[1]
         w, h = image.size