[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 (file)
--- 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: