From: Evan Shelhamer <shelhamer@imaginarynumber.net>
Date: Sat, 7 Mar 2015 00:23:43 +0000 (-0800)
Subject: [pycaffe] no need to squeeze output after #1970
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=fc35930e9dd366d87094213bf8a3c546edd2fefd;p=platform%2Fupstream%2Fcaffe.git

[pycaffe] no need to squeeze output after #1970

fix #2041 reported by @dgmp88
---

diff --git a/python/caffe/classifier.py b/python/caffe/classifier.py
index 94dd063..3d433ec 100644
--- a/python/caffe/classifier.py
+++ b/python/caffe/classifier.py
@@ -83,7 +83,7 @@ class Classifier(caffe.Net):
         for ix, in_ in enumerate(input_):
             caffe_in[ix] = self.transformer.preprocess(self.inputs[0], in_)
         out = self.forward_all(**{self.inputs[0]: caffe_in})
-        predictions = out[self.outputs[0]].squeeze(axis=(2,3))
+        predictions = out[self.outputs[0]]
 
         # For oversampling, average predictions across crops.
         if oversample: