[pycaffe] no need to squeeze output after #1970

shelhamer · shelhamer · commit fc35930e9dd3 · 2015-03-08T00:16:26.000-08:00
fix #2041 reported by @dgmp88
diff --git a/python/caffe/classifier.py b/python/caffe/classifier.py
@@ -83,7 +83,7 @@ def predict(self, inputs, oversample=True):
         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:
