process output

LouisFinner · Apr 2, 2018 · 4bbb335 · 4bbb335
1 parent 2993037
commit 4bbb335
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/predictor.py b/predictor.py
@@ -97,6 +97,7 @@ def restore(self, model_path):
     def predict(self, image):
         pos = self.sess.run(self.x_op, 
                     feed_dict = {self.x: image[np.newaxis, :,:,:]})
+        pos = np.squeeze(pos)
         return pos*self.MaxPos
 
     def predict_batch(self, images):